|
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
|
|