瀏覽代碼

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