Skip to content

Commit c066d17

Browse files
authored
Merge pull request #146 from glgh/doc-fix
[docs]: fix VQVAE training script path
2 parents 0ddc8c2 + 7373dab commit c066d17

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ Refer to the document [EVAL.md](docs/EVAL.md).
201201
To train VQVAE, run the script:
202202

203203
```
204-
scripts/train_vqvae.sh
204+
scripts/videogpt/train_videogpt.sh
205205
```
206206

207207
You can modify the training parameters within the script. For training parameters, please refer to [transformers.TrainingArguments](https://huggingface.co/docs/transformers/v4.38.2/en/main_classes/trainer#transformers.TrainingArguments). Other parameters are explained as follows:

docs/VQVAE.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ vqvae = VideoGPTVQVAE()
2121
To train the VQVAE model, you can use the `train_videogpt.sh` script. This script will train the model using the parameters specified in the script.
2222

2323
```bash
24-
bash scripts/train_videogpt.sh
24+
bash scripts/videogpt/train_videogpt.sh
2525
```
2626

2727
### Loading Pretrained Models

0 commit comments

Comments
 (0)