최강혁 / dddd

Go to a project
Toggle navigation Toggle navigation pinning
yunjey
eaa2d5ec 1 parent 4e56e6dc

name scope added

Inline Side-by-side
Showing 1 changed file with 2 additions and 0 deletions
......@@ -125,6 +125,7 @@ class DTN(object):
f_vars = [var for var in t_vars if 'content_extractor' in var.name]
# train op
with tf.name_scope('source_train_op'):
self.d_train_op_src = slim.learning.create_train_op(self.d_loss_src, self.d_optimizer_src, variables_to_train=d_vars)
self.g_train_op_src = slim.learning.create_train_op(self.g_loss_src, self.g_optimizer_src, variables_to_train=g_vars)
self.f_train_op_src = slim.learning.create_train_op(self.f_loss_src, self.f_optimizer_src, variables_to_train=f_vars)
......@@ -158,6 +159,7 @@ class DTN(object):
self.g_optimizer_trg = tf.train.AdamOptimizer(self.learning_rate)
# train op
with tf.name_scope('target_train_op'):
self.d_train_op_trg = slim.learning.create_train_op(self.d_loss_trg, self.d_optimizer_trg, variables_to_train=d_vars)
self.g_train_op_trg = slim.learning.create_train_op(self.g_loss_trg, self.g_optimizer_trg, variables_to_train=g_vars)
......