Skip to content

Commit

Permalink
Fix arg names in docs
Browse files Browse the repository at this point in the history
  • Loading branch information
AlbertDominguez authored Oct 16, 2024
1 parent 9fda402 commit 9e28e88
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/source/finetune.rst
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ You can fine-tune from an existing model by simply adding an argument to the CLI

.. code-block:: console
spotiflow-train /path/to/spots_data --save-dir /my/trained/model --pretrained-model general
spotiflow-train /path/to/spots_data -o /my/trained/model --finetune-from general
where `/path/to/pretrained/model` is the path to the directory containing the model you want to fine-tune. You can also pass other parameters to the training, such as the number of epochs, the learning rate, etc. For more information on the arguments allowed, see the documentation of the CLI command:

Expand Down Expand Up @@ -47,4 +47,4 @@ Finetuning a pre-trained model on a custom dataset is very easy. You can load th
)
Of course, you can also fine-tune from a model you have trained before. In that case, use the ``from_folder()`` method instead of ``from_pretrained()`` (see :ref:`index:Predicting spots in an image`).
All the information about training customization from :ref:`train:Customizing the training` applies here as well. However, note that you cannot change the model architecture when fine-tuning!
All the information about training customization from :ref:`train:Customizing the training` applies here as well. However, note that you cannot change the model architecture when fine-tuning!

0 comments on commit 9e28e88

Please sign in to comment.