Skip to content

Commit

Permalink
Fix typo (#269)
Browse files Browse the repository at this point in the history
  • Loading branch information
chauhang authored Oct 29, 2023
2 parents 6249331 + ad625a5 commit 23ef2ab
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/Dataset.md
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ def get_custom_dataset(dataset_config, tokenizer, split: str):
For an example `get_custom_dataset` you can look at the provided datasets in llama_recipes.datasets or [examples/custom_dataset.py](../examples/custom_dataset.py).
The `dataset_config` in the above signature will be an instance of llama_recipes.configs.dataset.custom_dataset with the modifications made through the command line.
The split signals wether to return the training or validation dataset.
The default function name is `get_custom_dataset` but this can be changes as described below.
The default function name is `get_custom_dataset` but this can be changed as described below.

In order to start a training with the custom dataset we need to set the `--dataset` as well as the `--custom_dataset.file` parameter.
```
Expand Down

0 comments on commit 23ef2ab

Please sign in to comment.