AttributeError: module 'torch.onnx' has no attribute 'TrainingMode'
时间: 2023-11-24 16:52:34 浏览: 110
AttributeError: module 'tensorflow.compat.v1' has no attribute '
根据提供的引用内容,可以看出出现了一个AttributeError,即模块'torch.onnx'没有'TrainingMode'属性。这个错误通常是由于使用了不兼容的PyTorch版本导致的。解决这个问题的方法是升级PyTorch版本或者降低使用的onnx版本。
以下是解决这个问题的步骤:
1.升级PyTorch版本。可以使用以下命令升级PyTorch:
```shell
pip install torch --upgrade
```
2.降低使用的onnx版本。可以使用以下命令降低onnx版本:
```shell
pip install onnx==<version>
```
其中,<version>是指要安装的onnx版本号。
阅读全文