Update ucf101_train.yaml

Vchitect · Feb 20, 2025 · f8980c5 · f8980c5
1 parent 72c41d2
commit f8980c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/configs/ucf101/ucf101_train.yaml b/configs/ucf101/ucf101_train.yaml
@@ -2,7 +2,7 @@
 dataset: "ucf101"
 
 data_path: "/path/to/datasets/UCF101/videos/"
-pretrained_model_path: "/path/to/pretrained/Latte/"
+pretrained_model_path: "/path/to/pretrained/Latte/" # Please download pre-trained models from https://huggingface.co/maxin-cn/Latte/tree/main
 
 # save and load
 results_dir: "./results"
@@ -39,4 +39,4 @@ num_classes: 101
 use_compile: False
 mixed_precision: False
 enable_xformers_memory_efficient_attention: False
-gradient_checkpointing: False
+gradient_checkpointing: False