gmalivenko / pytorch2keras

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

It's not working with Pytorch = 1.0 #58

Closed Golbstein closed 5 years ago

Golbstein commented 5 years ago

I've downgraded to 0.4.1 and it worked

gmalivenko commented 5 years ago

Hello @Golbstein. Yes, I plan to rewrite graph traverse function. It works for very simple models with PyTorch 1.0. If you'll send me a log with errors, I can fix it in future.

Golbstein commented 5 years ago

Actually I think the problem is with my machine...

When I run this code in my laptop it works (windows) but when I run it on my linux server it throws me "Segmentation fault (core dumped)" error. Do you know what's that?

gmalivenko commented 5 years ago

Are you sure that PyTorch and Tensorflow works on your machine? My library doesn't have C-backend or any other low-level operations with memory.

Golbstein commented 5 years ago

yes