Skip to content

GitLab

Unverified Commit 9e2fbc6b authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed bug in calculating `total_loss`

parent 4c75cd7d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_deeplab.py
View file @ 9e2fbc6b
...@@ -182,9 +182,9 @@ class PanopticDeeplabTask(base_task.Task): ...@@ -182,9 +182,9 @@ class PanopticDeeplabTask(base_task.Task):
loss_config.center_heatmap_loss_weight * instance_center_heatmap_loss + loss_config.center_heatmap_loss_weight * instance_center_heatmap_loss +
loss_config.center_offset_loss_weight * instance_center_offset_loss) loss_config.center_offset_loss_weight * instance_center_offset_loss)
total_loss = model_loss
if aux_losses: if aux_losses:
reg_loss = tf.add_n(aux_losses) total_loss += tf.add_n(aux_losses)
total_loss = model_loss + reg_loss
losses = { losses = {
'total_loss': total_loss, 'total_loss': total_loss,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment