Pytorch AttributeError: модуль 'torch' не имеет атрибута 'set_grad_enabled'

Question

AttributeError: модуль 'torch' не имеет атрибута 'set_grad_enabled'

Traceback (последний вызов был последним): файл "/home/thaqafi/PycharmProjects/transfer_learning/tutorial_transfer.py", строка 280, в

model_ft = train_model(model_ft, criterion, optimizer_ft, exp_lr_scheduler, num_epochs=25)

Файл "/home/thaqafi/PycharmProjects/transfer_learning/tutorial_transfer.py", строка 179, в train_model с torch.set_grad_enabled (True):

AttributeError: модуль 'torch' не имеет атрибута 'set_grad_enabled'

с использованием исходного кода веб-сайта Pytorch.

http://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

Answer 1

Насколько я знаю, эта функция не была доступна до версии 0.4.0, поэтому, вероятно, вы используете версию ниже 0.4.0 (например, 0.3.1.).

Попробуйте обновить PyTorch до версии 0.4.0. если возможно.