Toybrick

Couldn't not load resnet18.pt

jefferyzhang

版主

积分
13034
楼主
发表于 2020-2-20 11:26:44 | 显示全部楼层
pytorch version MUST equals 1.2
回复

使用道具 举报

jefferyzhang

版主

积分
13034
沙发
发表于 2020-2-24 17:20:50 | 显示全部楼层
nvnn 发表于 2020-2-24 08:22
Hi, pytorch version is 1.2.0. However still has some problem converting resnet18.pt
resnet18.pt **** ...

The resnet18 of torchvision version does't have softmax layer, aten::softmax is written by yourself?
You can try remove this op, and do softmax in 3399 side,
or try our fix version 1.3.1beta as below:
https://pan.baidu.com/s/1Kn2FGAdF_j3CMLNEsC3OPw
password:   rcds
回复

使用道具 举报

jefferyzhang

版主

积分
13034
板凳
发表于 2020-2-26 08:37:46 | 显示全部楼层
Can the version 1.3.1beta fix this problem?
回复

使用道具 举报

jefferyzhang

版主

积分
13034
地板
发表于 2020-2-26 10:54:45 | 显示全部楼层
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表