فهرست منبع

Use the old clone loss name instead

Neal Wu 8 سال پیش
والد
کامیت
289b439308
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  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