本代码为教程——《Pytorch模型训练实用教程》中配套代码;
《Pytorch模型训练实用教程》可通过如下方式获取:
代码在以下两种环境测试过:
- win10 64位 + python3.5 + pytorch==0.4.0
- mac + python3.6 + pytorch==0.4.1/ pytorch==1.0.0
第一步 安装各依赖包:
pip install -r requirements.txt
第二步 手动安装pytorch及torchvision:
均选择无gpu版本进行安装,进入官网选择相应的指令进行安装
https://pytorch.org/get-started/locally/
若发现任何问题和改进意见,请您随时联系我。
联系方式:yts3221@126.com
读者qq群:671103375
0.0.5:
- 1.6小节勘误,将36*36改为40*40;
- 2.3小节删除注释;
- 修改权值初始化杂谈中的理解错误;
- 全文代码缩进。