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

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