简体   繁体   中英

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.

You can convert your model to double by doing

model.double()

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

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM