pytorch中 model.cuda的作用
在pytorch中,即使是有GPU的機器,它也不會自動使用GPU,而是需要在程序中顯示指定。調用model.cuda(),可以將模型加載到GPU上去。這種方法不被提倡,而建議使用model.to(de ...
在pytorch中,即使是有GPU的機器,它也不會自動使用GPU,而是需要在程序中顯示指定。調用model.cuda(),可以將模型加載到GPU上去。這種方法不被提倡,而建議使用model.to(de ...