Похоже на , это , но OP заявляет, что его программа успешно работала на linux, в то время как моя проблема на Ubuntu. Я попытался понизить как факел, так и видение, и проблема не исчезла. Подозреваемая часть, которая вызывает проблему, показана ниже:
for epoch in range(max_epoch):
running_loss = 0.0
for image, label in train_loader:
optimiser.zero_grad()
output = model(image)
print("Label:{}\n".format(label))
print("Output is {}\n".format(output))
loss = criterion(output,label)
print("Loss is {}".format(loss.item()))
loss.backward()
print("Backward loss and now optimiser")
optimiser.step()
running_loss += loss.item()
print("Loss: {:.4f}".format(running_loss))
Полный код доступен здесь Вот подсказка, которую я получаю: Терминал