gmalivenko / pytorch2keras

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

Update conditions on pytorch versions #54

Closed EloiZalczer closed 5 years ago

EloiZalczer commented 5 years ago

Update conditions to support pytorch sub-versions between 0.4.1 and 1.0.0

Tested on Pytorch 0.4.1.post2 and 0.5.0a0. PLEASE CHECK THAT IT WORKS WITH 1.0.0 as I could not test it myself