search

gmalivenko / pytorch2keras

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

Hi i got this error: #62

Closed MNaseerSubhani closed 5 years ago

MNaseerSubhani commented 5 years ago

k_model = pytorch_to_keras(single_model, input_var, [(3, None, None,)], verbose=True)

1

gmalivenko commented 5 years ago

hello @MNaseerSubhani . You try to convert model located in CUDA memory, or try to pass a tensor located in CUDA as an argument. Please, make sure both model and arguments are in cpu memory.

MNaseerSubhani commented 5 years ago

Thank you. got it