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(zip(grads, child_variables_list))
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交