AttributeError: module ‘torch.onnx‘ has no attribute ‘set_training‘

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

你可能感兴趣的:(pytorch,pytorch,深度学习,python)