AttributeError: module 'torch.nn' has no attribute 'Mish'
时间: 2023-08-05 07:02:35 浏览: 611
开发errorjsp-手把手带你入门ssh项目
引用[1]中提到了解决"AttributeError: module 'torch.nn' has no attribute 'Mish'"错误的具体步骤。根据错误提示,这个错误是由于torch.nn模块中没有"Mish"属性引起的。为了解决这个问题,可以按照以下步骤进行操作:
1. 打开文件位置:/home/user/anaconda3/envs/chf_torch17_cp36/lib/python3.6/site-packages/torch_geometric/nn/dense/linear.py。
2. 在文件中找到以下代码段:
```python
if isinstance(self.weight, nn.parameter.UninitializedParameter):
self.weight = nn.Parameter(torch.Tensor(out_channels, in_channels))
nn.init.xavier_uniform_(self.weight)
```
将其替换为:
```python
if is_uninitialized_parameter(self.weight):
self.weight = nn.Parameter(torch.Tensor(out_channels, in_channels))
nn.init.xavier_uniform_(self.weight)
```
3. 在文件中找到以下代码段:
```python
if isinstance(self.bias, nn.parameter.UninitializedParameter):
self.bias = nn.Parameter(torch.Tensor(out_channels))
nn.init.zeros_(self.bias)
```
将其替换为:
```python
if is_uninitialized_parameter(self.bias):
self.bias = nn.Parameter(torch.Tensor(out_channels))
nn.init.zeros_(self.bias)
```
4. 保存文件并重新运行代码,应该就能够解决"AttributeError: module 'torch.nn' has no attribute 'Mish'"错误了。
希望这个解决方案能够帮助到你!
阅读全文