Deep Learning
[tensorflow] training specific variables
ddokkddokk
2017. 11. 14. 14:32
반응형
tvars = tf.trainable_variables()
g_vars = [var for var in tvars if 'g_' in var.name]
g_trainer = tf.train.AdamOptimizer(0.0001).minimize(g_loss, var_list=g_vars)