Skip to content

Commit

Permalink
Adding reference to plot_metrics and save_metrics in the fine tuning …
Browse files Browse the repository at this point in the history
…section of the readme
  • Loading branch information
albertodepaola committed Dec 26, 2023
1 parent d85cf35 commit 4a745cc
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,8 @@ All the parameters in the examples and recipes below need to be further tuned to

* Make sure to set the right path to the model in the [training config](src/llama_recipes/configs/training.py).

* To save the loss and perplexity metrics for evaluation, enable this by passing `--save_metrics` to the finetuning script. The file can be plotted using the [plot_metrics.py](./examples/plot_metrics.py) script, `python examples/plot_metrics.py --file_path path/to/metrics.json`

### Single GPU:

```bash
Expand Down

0 comments on commit 4a745cc

Please sign in to comment.