Tensorflow 只训练部分参数
child_variables_list = []
for trainable_var in model.trainable_variables:
if trainable_var.shape[-1] == 10:
child_variables_list.append(trainable_var)
grads = tape.gradient(loss, child_variables_list)
# w’=w-lr*grad
optimizers2.apply_gradients.
原创
2021-11-02 11:08:06 ·
556 阅读 ·
0 评论