Pytorch to(device)用法 ... 這兩行程式碼放在讀取資料之前。 ... 這行程式碼的意思是將所有最開始讀取資料時的tensor變數copy一份到device所指定的GPU上去, ...
確定! 回上一頁