Merge pull request #146 from glgh/doc-fix

[docs]: fix VQVAE training script path

Former-commit-id: 6ea832b958e6c8e5494a7b199874ffce061325aa
This commit is contained in:
lb203
2024-03-17 09:41:01 +08:00
committed by GitHub
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -201,7 +201,7 @@ Refer to the document [EVAL.md](docs/EVAL.md).
To train VQVAE, run the script:
```
scripts/train_vqvae.sh
scripts/videogpt/train_videogpt.sh
```
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:
+1 -1
View File
@@ -21,7 +21,7 @@ vqvae = VideoGPTVQVAE()
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.
```bash
bash scripts/train_videogpt.sh
bash scripts/videogpt/train_videogpt.sh
```
### Loading Pretrained Models