search

WZMIAOMIAO / deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.
GNU General Public License v3.0
23.25k stars 8.01k forks source link

swin_transformer export onnx : Process finished with exit code 136 (interrupted by signal 8: SIGFPE) #797

Open ddenglina opened 8 months ago

ddenglina commented 8 months ago 
dummy_input = torch.rand((1, 3, 224, 224))
torch.onnx.export(model,
                      dummy_input,
                      save_onnx_path,
                      export_params=True,
                      verbose=True,
                      do_constant_folding=True,
                      opset_version=11)

export onnx error: Process finished with exit code 136 (interrupted by signal 8: SIGFPE)