CODEWITHSUNDEEP
pythonmachine-learningkeraspytorch
Samed Mukush
Samed Mukush

Reputation: 31

Saving model in pytorch and keras

I have trained model with keras and saved in with the help of pytorch. Will it cause any problems in the future. As far as I know the only difference between them is Keras saves its model's weights as doubles while PyTorch saves its weights as floats.

Upvotes: 0

Views: 263

Answers (1)

Sagnik Mukherjee
Sagnik Mukherjee

Reputation: 106

You can convert your model to double by doing

model.double()

Note that after this, you will need your input to be DoubleTensor.

Upvotes: 1

Related Questions