首页 文章

Pytorch AttributeError:模块'torch'没有属性'set_grad_enabled'

提问于
浏览
0

AttributeError:模块'torch'没有属性'set_grad_enabled'

回溯(最近一次调用最后一次):文件“/home/thaqafi/PycharmProjects/transfer_learning/tutorial_transfer.py”,第280行,在

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

使用torch.set_grad_enabled(True)在train_model中输入文件“/home/thaqafi/PycharmProjects/transfer_learning/tutorial_transfer.py”,第179行:

AttributeError:模块'torch'没有属性'set_grad_enabled'

使用源代码形式Pytorch webiste .

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

1 回答

  • 1

    据我所知,此功能在版本0.4.0之前不可用,因此您可能使用的版本低于0.4.0(例如0.3.1 . ) .

    尝试将PyTorch更新到0.4.0版 . 如果可能的话 .

相关问题