diff options
author | Viveka Kulharia <vivkul@iitk.ac.in> | 2016-03-02 10:37:56 +0530 |
---|---|---|
committer | Viveka Kulharia <vivkul@iitk.ac.in> | 2016-03-02 10:37:56 +0530 |
commit | 4f4d0209495aa858de4ea7d71b3aea9196b14a05 (patch) | |
tree | 1c290b696845a76cd0fe43c9df5b0d7ac93950d6 /examples | |
parent | 93af70e464e973e79d8b9a12c5431c7c209d4ec9 (diff) | |
download | caffeonacl-4f4d0209495aa858de4ea7d71b3aea9196b14a05.tar.gz caffeonacl-4f4d0209495aa858de4ea7d71b3aea9196b14a05.tar.bz2 caffeonacl-4f4d0209495aa858de4ea7d71b3aea9196b14a05.zip |
minor mistakes removed
Diffstat (limited to 'examples')
-rw-r--r-- | examples/finetune_flickr_style/readme.md | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/finetune_flickr_style/readme.md b/examples/finetune_flickr_style/readme.md index 4e9d41f1..9ba4c921 100644 --- a/examples/finetune_flickr_style/readme.md +++ b/examples/finetune_flickr_style/readme.md @@ -14,9 +14,9 @@ Let's fine-tune the BVLC-distributed CaffeNet model on a different dataset, [Fli ## Explanation The Flickr-sourced images of the Style dataset are visually very similar to the ImageNet dataset, on which the `bvlc_reference_caffenet` was trained. -Since that model works well for object category classification, we'd like to use it architecture for our style classifier. +Since that model works well for object category classification, we'd like to use this architecture for our style classifier. We also only have 80,000 images to train on, so we'd like to start with the parameters learned on the 1,000,000 ImageNet images, and fine-tune as needed. -If we give provide the `weights` argument to the `caffe train` command, the pretrained weights will be loaded into our model, matching layers by name. +If we provide the `weights` argument to the `caffe train` command, the pretrained weights will be loaded into our model, matching layers by name. Because we are predicting 20 classes instead of a 1,000, we do need to change the last layer in the model. Therefore, we change the name of the last layer from `fc8` to `fc8_flickr` in our prototxt. |