将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 ·
1047 阅读 ·
2 评论