pytorch中 model.cuba的作用是什麼？

在pytorch中，即使是有GPU的機器，它也不會自動使用GPU，而是需要在程序中顯示指定。調用model.cuda()，可以將模型加載到GPU上去。這種方法不被提倡，而建議使用model.to(device)的方式，這樣可以顯示指定需要使用的計算資源，特別是有多個GPU的情況下。