PyTorch报”TypeError: max() got an unexpected keyword argument ‘dim’ “的原因以及解决办法

目录

• 手动替换参数
• 更新PyTorch版本

PyTorch 中的 "TypeError: max() got an unexpected keyword argument 'dim' " 错误通常是因为该版本的 PyTorch 版本不支持 'dim' 参数导致的。在过去的版本中， 'dim' 参数被称为 'dimention' 参数。

由于 'dim' 参数是 PyTorch 1.x 版本中的新增功能，因此在旧版本的 PyTorch 中使用 'dim' 参数会导致错误。

解决此问题的方法很简单，只需将 'dim' 参数替换为 'dimention' 参数即可。以下是解决方案：

手动替换参数

手动更改代码中的 'dim' 参数，将其替换为 'dimention'，如下所示：

output, _ = torch.max(inputs, dimention=1)

更新PyTorch版本

另一种解决该错误的方法是将 PyTorch 更新到最新版本，即PyTorch 1.8及以上版本，因为这些版本已经支持 'dim' 参数。更新 PyTorch 的最简单方法是通过 pip 命令进行更新，如下所示：

!pip install torch==1.8.0

或者在 Anaconda 中使用以下命令：

conda install pytorch==1.8.0

总之，以上两种方法都可以解决 "TypeError: max() got an unexpected keyword argument 'dim' " 错误。选择哪种方法取决于你个人喜好和对代码的修改程度。如果你的 PyTorch 版本较旧，可能最好更新到最新版本，否则建议手动替换参数。