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):
# Some layer like tf.layers.batch_norm need this:
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
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