search

sczhou / STFAN

[ICCV 2019] Spatio-Temporal Filter Adaptive Network for Video Deblurring
MIT License
172 stars 31 forks source link

onnx #17

Open fengchenTM opened 2 years ago

fengchenTM commented 2 years ago

hi, when I export model to onnx format, I meet a problem File "/root/anaconda3/lib/python3.8/site-packages/torch/onnx/init.py", line 316, in export return utils.export(model, args, f, export_params, verbose, training, File "/root/anaconda3/lib/python3.8/site-packages/torch/onnx/utils.py", line 107, in export _export(model, args, f, export_params, verbose, training, input_names, output_names, File "/root/anaconda3/lib/python3.8/site-packages/torch/onnx/utils.py", line 737, in _export proto, export_map, val_use_external_data_format = graph._export_onnx( RuntimeError: ONNX export failed: Couldn't export Python operator KernelConv2DFunction