windows+caffe+mnist識別

1.   打開examples\mnist目錄下的lenet_solver.prototxt文件,可以看到solver_mode: GPU這一選項,可以設置爲GPU或者CPU net和snapshot_prefix  設置爲絕對路徑 2.   打開examples\mnist目錄下的lenet_train_test.prototxt文件,修改第一個和第二個layer的source爲第一步生成的訓練
相關文章
相關標籤/搜索