Skip to content

Commit

Permalink
Merge pull request #4024 from achalddave/finetune-flickr-style-tutorial
Browse files Browse the repository at this point in the history
[docs] explicitly point out -weights flag in fine-tuning tutorial
  • Loading branch information
shelhamer committed Apr 20, 2016
2 parents 5166583 + bd76210 commit faba632
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 faba632

Please sign in to comment.