Tensorflow v1.14中的梯度剪裁错误

2024-03-29 05:51:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我在训练过程中使用渐变剪辑,使用下面的代码

variables = tf.trainable_variables()
enc_vars = [v for v in variables if v.name.startswith('encoder')]

enc_opt = tf.train.AdamOptimizer()
gvs = enc_opt.compute_gradients(loss, var_list=enc_vars)
capped_gvs = [(tf.clip_by_value(grad, -10., 10.), var) for grad, var in gvs]
train_enc = enc_opt.apply_gradients(capped_gvs)

我已经检查过enc_varsgvs都不是空的。但是我在上面代码的第六行得到一个错误

ValueError: None values not supported.

我做错什么了?提前谢谢。你知道吗


Tags: 代码infor过程vartftrainvars