TensorFlow Pytorch 模型读取相关
TensorFlow Pytorch 模型读取相关TensorflowTensorflow载入部分权重基本步骤适用方法Tensorflow模型转Pytorch
Tensorflow
Tensorflow载入部分权重
基本步骤
为layer命名,加入name=‘layer_name’
model.load_weights(’…h5’, by_name=True)
适用方法
网络中间的某些层不载入权重
model.layers[id].name = 'layer_name_2' //重命名不想载入权重的
原创
2020-11-26 11:41:51 ·
749 阅读 ·
0 评论