Skip to content

Commit 9a56c41

Browse files
authored
added missing word and corrected spelling (meta-llama#707)
2 parents cc84da2 + 3f883f6 commit 9a56c41

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

recipes/quickstart/finetuning/finetune_vision_model.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,6 @@ In order to use a custom dataset, please follow the steps below:
2828

2929
1. Create a new dataset python file under `recipes/quickstart/finetuning/dataset` folder.
3030
2. In this python file, you need to define a `get_custom_dataset(dataset_config, processor, split, split_ratio=0.9)` function that handles the data loading.
31-
3. In this python file, you need to define a `get_data_collator(processor)` that returns a custom data collator that can be used by the Pytorch Data Loader.
31+
3. In this python file, you need to define a `get_data_collator(processor)` class that returns a custom data collator that can be used by the Pytorch Data Loader.
3232
4. This custom data collator class must have a `__call__(self, samples)` function that converts the image and text samples into the actual inputs that vision model expects.
33-
5. Run the `torchrun` commend from above section, please change the `--custom_dataset.file` to the new dataset python file, adjust the learning rate accordingly.
33+
5. Run the `torchrun` command from above section, please change the `--custom_dataset.file` to the new dataset python file, adjust the learning rate accordingly.

0 commit comments

Comments
 (0)