rwightman/gen-efficientnet-pytorch

torch2onnx Segmentation fault (core dumped)

vicwer opened this issue · 1 comments

vicwer commented

Hi,
I want to convert efficientnet_b0.pth to efficientnet_b0.onnx, when I runpython onnx_export.py --model=efficientnet_b0 res.onnx --checkpoint=../efficientnet_b0.pth:

==> Creating PyTorch efficientnet_b0 model
=> Loading checkpoint '../efficientnet_b0.pth'
=> Loaded checkpoint '../efficientnet_b0.pth'
==> Exporting model to ONNX format at 'res.onnx'
==> Loading and checking exported model from 'res.onnx'
Segmentation fault (core dumped)

what should I do?

vicwer commented

onnx==1.5.0, it works.