本代码为教程——《Pytorch模型训练实用教程》中配套代码;
《Pytorch模型训练实用教程》可通过如下方式获取:
- https://github.com/tensor-yu/PyTorch_Tutorial/tree/master/Data
- QQ群:671103375(已满) 773031536
代码在以下两种环境测试过:
- 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 (已满)
二群:773031536 (已满)
三群:514974779
0.0.5:
- 1.6小节勘误,将36*36改为40*40;
- 2.3小节删除注释;
- 修改权值初始化杂谈中的理解错误;
- 全文代码缩进。