Browse Source

Use the old clone loss name instead

Neal Wu 8 năm trước cách đây
mục cha
commit
289b439308
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      slim/deployment/model_deploy.py

+ 1 - 1
slim/deployment/model_deploy.py

@@ -232,7 +232,7 @@ def _gather_clone_loss(clone, num_clones, regularization_losses):
       sum_loss = tf.add_n(all_losses)
   # Add the summaries out of the clone device block.
   if clone_loss is not None:
-    tf.summary.scalar('clone_loss', clone_loss)
+    tf.summary.scalar(clone.scope + '/clone_loss', clone_loss)
   if regularization_loss is not None:
     tf.summary.scalar('regularization_loss', regularization_loss)
   return sum_loss