转载:https://blog.csdn.net/gaoyueace/article/details/79079068
例如:
#在名字为ae的命名空间内创建变量
with tf.variable_scope('ae'):
Y_pred, Y_pred_modi = self.ae_u(X)
with tf.variable_scope('dis'):
XY_real_pair = self.dis(X, Y)
with tf.variable_scope('dis',reuse=True):
XY_fake_pair = self.dis(X, Y_pred)
with tf.variable_scope('dis',reuse=True):
XY_fake_intep = self.dis(X, interpolates)
ae_var = [var for var in tf.trainable_variables() if var.name.startswith('ae')]
dis_var = [var for var in tf.trainable_variables() if var.name.startswith('dis')]
ae_g_optim = tf.train.AdamOptimizer(learning_rate=lr, beta1=0.9, beta2=0.999, epsilon=1e-8).minimize(ae_gan_g_loss, var_list=ae_var)
dis_optim = tf.train.AdamOptimizer(learning_rate=lr,beta1=0.9,beta2=0.999,epsilon=1e-8).minimize(gan_d_loss_gp,var_list=dis_var)