1.在PyTorch 1.6版本中set_training变成了select_model_mode_for_export,改一下。
2.降低版本pip install torch1.5.1 torchvision0.6.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
AttributeError: module ‘torch.onnx‘ has no attribute ‘set_training‘
最新推荐文章于 2024-01-26 09:32:52 发布