Caffe mnist数据集训练

作者: SnailTyan | 来源:发表于2016-12-05 18:12 被阅读70次

    文章作者:Tyan
    博客:noahsnail.com  |  CSDN  |  简书

    本文是使用基于docker的caffe进行实验。docker启动过程请看docker相关的博客。训练过程如下:

    # 进入caffe目录
    $:/workspace# cd /opt/caffe/
    
    # 下载mnist数据集
    $:/opt/caffe# ./data/mnist/get_mnist.sh 
    
    # 将mnist数据集转换成lmdb
    $:/opt/caffe# sh examples/mnist/create_mnist.sh
    
    # 修改examples/mnist/lenet_solver.prototxt
    将GPU改为CPU
    
    # 运行examples/mnist/train_lenet.sh
    $:/opt/caffe# sh examples/mnist/train_lenet.sh
    

    相关文章

      网友评论

        本文标题:Caffe mnist数据集训练

        本文链接:https://www.haomeiwen.com/subject/xrclmttx.html