NameError: name ‘nn’ is not defined

1、报错界面

2、分析原因
使用Pytorch框架，需引入torch和torch.nn（或只是nn），这是两个主要的 PyTorch 包：

import torch
import torch.nn as nn

需要包含这两行，因为如果您只设置第二行，如果torch未导入包，则可能无法正常工作。

可以help(torch.nn) 确认这一点。

3、解决方案
在.py 文件导入torch包进行修复即可：

import torch
import torch.nn as nn