大概修改的地方包括将数据的形式变成GPU 能读的形式, 然后将CNN 也变成GPU 能读的形式. 做法就是在后面加上 .cuda(), 很简单. ... test_data = torchvision.datasets.MNIST ...
確定! 回上一頁