deep-learning-for-image-processing icon indicating copy to clipboard operation
deep-learning-for-image-processing copied to clipboard

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

Open ddenglina opened this issue 1 year ago • 0 comments

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)

ddenglina avatar Mar 21 '24 06:03 ddenglina