Skip to content

Commit bd76210

Browse files
committed
Explicitly point out -weights flag in tutorial
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.
1 parent 045e5ba commit bd76210

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

examples/finetune_flickr_style/readme.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,11 @@ The prototxts in this example assume this, and also assume the presence of the I
5757

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

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

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

0 commit comments

Comments
 (0)