Merge pull request #146 from glgh/doc-fix
[docs]: fix VQVAE training script path Former-commit-id: 6ea832b958e6c8e5494a7b199874ffce061325aa
This commit is contained in:
@@ -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
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user