虛擬機Ubuntu 16.04 64位 Caffe 訓練LeNet 數據集MNIST

一、準備工作 1.準備數據集之前請先切換到root模式,具體指令爲: $ sudo su 然後輸入自己設定的密碼 2.準備數據集,具體指令爲: $ cd caffe/ $ ./data/mnist/get_mnist.sh $ ./examples/mnist/create_mnist.sh 二、訓練模型 1.因爲下載的訓練參數默認是GPU,我們需要將其改爲CPU,否則報錯,輸入以下指令: $ c
相關文章
相關標籤/搜索