module 'torch.nn' has no attribute 'Moudule'
时间: 2023-12-18 22:29:54 浏览: 169
PyTorch里面的torch.nn.Parameter()详解
3星 · 编辑精心推荐
`module 'torch.nn' has no attribute 'Moudule'`错误提示表明在torch.nn模块中没有名为'Moudule'的属性。这个错误可能是由于拼写错误导致的。正确的属性名应该是'Module',而不是'Moudule'。请检查你的代码并确保正确拼写了属性名。
以下是一个示例代码,展示了如何正确导入torch.nn模块并使用Module类:
```python
import torch
import torch.nn as nn
class LinearModel(nn.Module):
def __init__(self, ndim):
super(LinearModel, self).__init__()
self.ndim = ndim
self.weight = nn.Parameter(torch.randn(ndim, 1))
self.bias = nn.Parameter(torch.randn(1))
def forward(self, x):
return x.mm(self.weight) + self.bias
lm = LinearModel(5)
```
请注意,代码中的`super(LinearModel, self).__init__()`用于调用父类的构造函数,确保正确初始化模型。另外,`nn.Parameter`用于定义可训练的参数。
阅读全文