错误：
RuntimeError: CUDA error: out of memory
原因：网络预训练模型与Pytorch版本不匹配
参考：https://blog.csdn.net/eefresher/article/details/99979056?utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1.no_search_link&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1.no_search_link
感谢大佬！
解决方法：
