Pytorch AttributeError:模块'火炬'没有属性' set_grad_enabled'

时间:2018-04-30 09:30:13

标签: pytorch

属性错误:模块'火炬'没有属性' 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)

文件  " /home/thaqafi/PycharmProjects/transfer_learning/tutorial_transfer.py" ;,第179行,在train_model中     与torch.set_grad_enabled(True):

属性错误:模块'火炬'没有属性' set_grad_enabled'

使用源代码形式Pytorch webiste。

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

1 个答案:

答案 0 :(得分:1)

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

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