参考文档
- http://zh.gluon.ai/index.html 动手学深度学习-Gluon官方教程中文翻译版
- https://mxnet.incubator.apache.org/install/index.html
- http://blog.csdn.net/mumoDM/article/details/79502848 找不到libcusolver.so.9.0的解决方法
- http://blog.csdn.net/panda1942/article/details/50923006 mxnet深度学习实战:跑自己的数据实验和一些问题总结
- http://blog.csdn.net/shuzfan/article/details/50037273 mxnet学习记录
- http://blog.csdn.net/u011995719/article/details/77884728 Resnet-18-训练实验-warm up操作
安装Mxnet GPU训练版本
准备训练数据集
cd incubator-mxnet
./bin/im2rec /opt/xyl/data/mmr/train.txt /opt/xyl/data/mmr/ /opt/xyl/data/mmr/train.rec quality=95 resize=256
编写训练脚本
进行训练
注意事项:当执行sudo python train.py时出现找不到libcusolver.so.9.0的错误,可以通过执行
sudo ldconfig /usr/local/cuda/lib64 来解决
- [x] mxnet batch_size的问题
- [x] mxnet 图片增广
- [x] mxnet 均值文件
- [x] mxnet模型保存,loss显示
- [x] 从之前某个训练epoch开始继续训练