torch踩坑

报错

NameError: name 'nn' is not defined

解决方法：加入此语句,定义nn

import torch.nn as nn

初始化优化器时：

# 初始化优化器
optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9)

报错

NameError: name 'optim' is not defined

解决方法：
改为

# 初始化优化器
optimizer = torch.optim.SGD(model.parameters(), lr=0.001, momentum=0.9)