pytorch
BrightHao_zi
天道酬勤
展开
-
pytorch加载模型时报错 AttributeError: Can‘t get attribute ‘CNN‘ on <module ‘__main__‘>
pytorch加载训练好的模型时报错:解决方法:在加载模型的这个文件中,将模型的定义代码复制过来。参考博客:pytorch 在加载模型时出现的错误原创 2021-02-03 19:46:53 · 6182 阅读 · 0 评论 -
Pytorch中backward()的思考记录
在pytorch中,只能对标量使用backward。如果对向量进行backward,则会报错:import torchx=torch.tensor([2,3,4],dtype=torch.float,requires_grad=True)print(x)y=x*2print(y)# z=y.mean()# z.backward()y.backward()print(x.requires_grad)print(x.grad)以上代码运行:RuntimeError: grad can原创 2020-09-29 19:26:12 · 1061 阅读 · 0 评论