花费 25 ms
pytorch中 model.cuda的作用

在pytorch中,即使是有GPU的机器,它也不会自动使用GPU,而是需要在程序中显示指定。调用model.cuda(),可以将模型加载到GPU上去。这种方法不被提倡,而建议使用model.to(de ...

Wed Oct 28 22:18:00 CST 2020 0 2057

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM