- 博客(1)
- 收藏
- 关注
RSS订阅原创 将TF2.0 Tensorflow2.X 模型参数 转换成PyTorch模型的参数
首先定义你自己的TF模型，并加载训练好的模型文件（不加载也可以）classMyModel1:TFmodel = MyModel()TFmodel.load_weights('./training_checkpoint_265.h5', by_name=True)然后定义一个PyTorch模型（注意，这里的Pytorch模型结构必须和TF模型结构完全一样）class MyModel2(nn.Module):PyTorchModel = MyModel2()然后就可以..
2021-04-28 19:57:13
1219
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人