请教下keras的CNN网络怎么转换为Torch版本的

img


请教下怎么把一个keras的网络模型转换为cnn的模型 数据的shape为(3919,1000,2)

我当时是手动一层一层的对应输出,用pytorch复现了一遍,不过结果会有误差,自己随机一个输入,每次输出大小,用pytorch再搞一边,其他方法都不靠谱,还有就是论文中的话github可能有人复现了,多找找,backbone啥的pytorch自带

可以参考下这个

根据他定义的模型的各个层在pytorch里面调用相应的api实现相应的层就行了,但需要题主了解一点pytorch模型构建的知识和语法

不用转换,你直接找你的网络对应的torch版本,比如vgg16这些,很多网上都有torch的版本的

参考:
https://gereshes.com/2019/06/24/how-to-transfer-a-simple-keras-model-to-pytorch-the-hard-way/

import torch.nn as nn
下面是代码

您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!
PS:问答VIP年卡 【限时加赠:IT技术图书免费领】,了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632