onnx / tutorials

Tutorials for creating and using ONNX models
Apache License 2.0
3.39k stars 629 forks source link

Going from PyTorch to Keras #231

Open AzazelHD opened 3 years ago

AzazelHD commented 3 years ago

Hi. I am following this guide to achieve exporting a pytorch model (this one https://github.com/ypeleg/U-2-Net-Keras , under Usage for salient object detection).

My plan is to: PyTorch -> Onnx -> Keras.

But going from PyTorch -> Onnx i get this error: AttributeError: 'collections.OrderedDict' object has no attribute 'training'

Please, can someone help me?