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