让 slim 知道额外的 loss 并处理 losses

最新推荐文章于 2022-07-18 16:37:52 发布

Gallant Hu

最新推荐文章于 2022-07-18 16:37:52 发布

阅读量142

点赞数

分类专栏：机器学习二

本文链接：https://blog.csdn.net/weixin_42108090/article/details/111874019

版权

机器学习二专栏收录该内容

18 篇文章 1 订阅

订阅专栏

# Load the images and labels.
images, scene_labels, depth_labels, pose_labels = ...

# Create the model.
scene_predictions, depth_predictions, pose_predictions = CreateMultiTaskModel(images)

# Define the loss functions and get the total loss.
classification_loss = slim.losses.softmax_cross_entropy(scene_predictions, scene_labels)
sum_of_squares_loss = slim.losses.sum_of_squares(depth_predictions, depth_labels)
pose_loss = MyCustomLossFunction(pose_predictions, pose_labels)
slim.losses.add_loss(pose_loss) # Letting TF-Slim know about the additional loss.

# The following two ways to compute the total loss are equivalent:
regularization_loss = tf.add_n(slim.losses.get_regularization_losses())
total_loss1 = classification_loss + sum_of_squares_loss + pose_loss + regularization_loss

# (Regularization Loss is included in the total loss by default).
total_loss2 = slim.losses.get_total_loss()