机器学习日记
研究生,学习日记
kin55
万事可期!!!
展开
-
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*gradoptimizers2.apply_gradients.原创 2021-11-02 11:08:06 · 495 阅读 · 0 评论 -
同时安装tensorflowl和pytorch和torch-geometric
其实根本不用换源pip install tensorflow==2.1 -i https://pypi.tuna.tsinghua.edu.cn/simple原创 2020-08-07 22:01:22 · 882 阅读 · 0 评论