deep-learning-for-image-processing
deep-learning-for-image-processing copied to clipboard
swin_transformer export onnx : Process finished with exit code 136 (interrupted by signal 8: SIGFPE)
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)