Skip to content

Commit

Permalink
Explicitly point out -weights flag in tutorial
Browse files Browse the repository at this point in the history
The -weights flag is somewhat easy to miss as it's only in one command,
but is the crucial thing that anyone searching for 'how to finetune' is
looking for. Hopefully this more clearly points out the '-weights' flag,
which might otherwise be overlooked in this tutorial.
  • Loading branch information
achalddave committed Apr 20, 2016
1 parent 045e5ba commit bd76210
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion examples/finetune_flickr_style/readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,11 @@ The prototxts in this example assume this, and also assume the presence of the I

We'll also need the ImageNet-trained model, which you can obtain by running `./scripts/download_model_binary.py models/bvlc_reference_caffenet`.

Now we can train! (You can fine-tune in CPU mode by leaving out the `-gpu` flag.)
Now we can train! The key to fine-tuning is the `-weights` argument in the
command below, which tells Caffe that we want to load weights from a pre-trained
Caffe model.

(You can fine-tune in CPU mode by leaving out the `-gpu` flag.)

caffe % ./build/tools/caffe train -solver models/finetune_flickr_style/solver.prototxt -weights models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel -gpu 0

Expand Down

0 comments on commit bd76210

Please sign in to comment.