属性错误:模块'火炬'没有属性' 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
答案 0 :(得分:1)
据我所知,此功能在版本0.4.0之前不可用,因此您可能使用的版本低于0.4.0(例如0.3.1。)。
尝试将PyTorch更新到0.4.0版。如果可能的话。