From 83cf1dbd8e8c4815a8788938722b6a55404ac0d6 Mon Sep 17 00:00:00 2001 From: gl Date: Sat, 16 Mar 2024 12:12:37 -0700 Subject: [PATCH] [docs]: fix VQVAE script path Former-commit-id: d5f5c681b8c50622a84256881d193d5803ff3c28 --- README.md | 2 +- docs/VQVAE.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 83f90236a..91258fb5c 100644 --- a/README.md +++ b/README.md @@ -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: diff --git a/docs/VQVAE.md b/docs/VQVAE.md index 823c56bab..a1af36364 100644 --- a/docs/VQVAE.md +++ b/docs/VQVAE.md @@ -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