model = ResNet(ResidualBlock, [2, 2, 2, 2]).to(device) # 设置损失函数和优化器 criterion = nn.CrossEntropyLoss() optimizer = torch.optim.
確定! 回上一頁