How to test on gpu

[jay985735639]

hi! my friend,How to test on gpu?

[Developer-Zer0]

Hi!! As our model was trained on CPU, you will have to change the weights to GPU while loading the model. You will have to make some changes in the code for that.

You can refer to this link for reference: https://pytorch.org/tutorials/recipes/recipes/save_load_across_devices.html

Hope this was helpful

[jay985735639]

Hi!! As our model was trained on CPU, you will have to change the weights to GPU while loading the model. You will have to make some changes in the code for that.

You can refer to this link for reference: https://pytorch.org/tutorials/recipes/recipes/save_load_across_devices.html

Hope this was helpful

Thanks for your help