Skip to content

ResNet50 Training

Hiroki Naganuma edited this page Jul 12, 2017 · 1 revision

NV-Caffe 0.16ではcaffe binaryに実行ファイルは統合されている experimental-fp16 branchの頃は caffe?fp16 binaryがありそちらを指定した(trainオプションは選べなかった)

例えば以下のようなコマンドでtrainingを実行できる

$CAFFE_HOME/build/tools/caffe train -solver models/resnet50/solver_fp16.prototxt -gpu 0,1,2,3
Clone this wiki locally