我是 PyTorch 和神经网络的真正新手。本周我开始研究这些主题，我的导师给了我一个代码以及一些处理该代码的任务。但他给我的代码不起作用。我一整天都在尝试解决这个问题，但没有结果。因为我不知道 NN 和 PyTorch 的背景，所以很难理解这个问题。需要你的帮助。谢谢 ！
import torch
import numpy as np
import torchvision.datasets as datasets
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
from torchsummary import summary
#DEFINE YOUR DEVICE
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
print(device) #if cpu, go Runtime-> Change runtime type-> Hardware accelerator GPU -> Save -> Redo previous steps
#DOWNLOAD DATASET
train_data = datasets.Omniglot('./data', background=True, download = True, transform = transforms.ToTensor())
test_data = datasets.Omniglot('./data',background = False, download = True, transform = transforms.ToTensor()) …