소스 검색

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