将Tensorflow模型转化为Pytorch模型的方法总结
最近做的一个事情需要将已经训练好的TF模型迁移到Pytorch上使用,于是在此记录一下。一、方法一:该方法比较“原始”,大致的做法是:1.读取TF模型-->2.获取参数名和值构成的字典-->3.构建Pytorch模型-->5.使用已经获取的字典对Pytorch模型参数进行更新-->6.保存为.pth文件具体的实现上参考:https://www.cnblogs.c...
原创
2020-04-28 10:06:21 ·
20106 阅读 ·
8 评论