gmalivenko / pytorch2keras

PyTorch to Keras model convertor
https://pytorch2keras.readthedocs.io/en/latest/
MIT License
857 stars 143 forks source link

Update tracing module #84

Closed gmalivenko closed 5 years ago

gmalivenko commented 5 years ago

Due to new PyTorch updates the only way to proper handle weights and layer names at the same time - to use ONNX export.

From the next version of the converter, tracing module will be replaced. Hopefully it will open new features - converting modules such as ModuleList, converting recurrent networks and so on. Stay tuned.

The new tracing module will be placed to the separated repository https://github.com/nerox8664/onnx2keras and used as a library after implementation.

Progress: