Crash when trying to export PyTorch model to ONNX: forward() missing 1 required positional argument

后端 未结 0 1816
我在风中等你
我在风中等你 2021-02-18 22:56

I\'m trying to convert pyTorch model to onnx like this:

torch.onnx.export(
  model=modnet.module,
  args=example_input, 
  f=ONNX_PATH, # where should it be saved         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题