ResNet50 Training - rioyokotalab/caffe GitHub Wiki

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