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
所有评论(0)