本代码为教程——《Pytorch模型训练使用教程》中配套代码;
《Pytorch模型训练使用教程》可通过如下方式获取:
代码在以下两种环境测试过:
- win10 64位 + python3.5 + pytorch==0.4.0
- mac + python3.6 + pytorch==0.4.1
第一步 安装各依赖包:
pip install -r requirements.txt
第二步 手动安装pytorch及torchvision:
均选择无gpu版本进行安装,进入官网选择相应的指令进行安装
https://pytorch.org/get-started/locally/
若发现任何问题,小bug,改进意见,请您随时联系我,以进行改进。
联系方式:[email protected]
读者qq群:671103375