命令、语句疑难及相关理解

xiaoxiao2021-02-27  470

1. var_list = [    v for v in tf.trainable_variables() if v.name.split('/')[0] in train_layers] 

for v in tf.trainable_variables() 先放入v中; 然后if v.name.split('/')[0] in train_layers从中筛选符合条件的放入v中,相当于保留下来得到var_list的值

tf.trainable_variables() 返回模型中所训练的参数,可以用 for v in tf.trainable_variables(): print(v) 查看里面的内容,或者用print(tf.trainable_variables())查看

转载请注明原文地址: https://www.6miu.com/read-1603.html

最新回复(0)