Commit b89aa8ab authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

put a name scpoe for train parts

parent 8606bd03
...@@ -100,7 +100,7 @@ class Regressor(estimator.Estimator): ...@@ -100,7 +100,7 @@ class Regressor(estimator.Estimator):
# Some layer like tf.layers.batch_norm need this: # Some layer like tf.layers.batch_norm need this:
update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS) update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS)
with tf.control_dependencies(update_ops): with tf.control_dependencies(update_ops), tf.name_scope('train'):
# Calculate Loss # Calculate Loss
self.loss = self._get_loss( self.loss = self._get_loss(
......
Supports Markdown
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