参考文档

  1. http://zh.gluon.ai/index.html 动手学深度学习-Gluon官方教程中文翻译版
  2. https://mxnet.incubator.apache.org/install/index.html
  3. http://blog.csdn.net/mumoDM/article/details/79502848 找不到libcusolver.so.9.0的解决方法
  4. http://blog.csdn.net/panda1942/article/details/50923006 mxnet深度学习实战:跑自己的数据实验和一些问题总结
  5. http://blog.csdn.net/shuzfan/article/details/50037273 mxnet学习记录
  6. 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开始继续训练

results matching ""

    No results matching ""