tutorials icon indicating copy to clipboard operation
tutorials copied to clipboard

Going from PyTorch to Keras

Open AzazelHD opened this issue 4 years ago • 0 comments

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?

AzazelHD avatar Jan 03 '21 00:01 AzazelHD