[feat]: Add adaptive fps dataloader and remove redundant code (#41)

No checkout mochi
This commit is contained in:
rlsu9
2024-11-06 19:52:48 -08:00
parent 035ba5f5cf
commit 44bda9f8a3
4 changed files with 17 additions and 115 deletions
+1 -2
View File
@@ -1,7 +1,6 @@
# export WANDB_MODE="offline"
GPU_NUM=8
MODEL_PATH="/ephemeral/hao.zhang/outputfolder/ckptfolder/mochi_diffuser"
MOCHI_DIR="/ephemeral/hao.zhang/resourcefolder/mochi/mochi-1-preview"
DATA_MERGE_PATH="/ephemeral/hao.zhang/resourcefolder/Mochi-Synthetic-Data-BW-Finetune/merge.txt"
OUTPUT_DIR="./data/BW-Finetune-Synthetic-Data_test"
@@ -12,7 +11,7 @@ torchrun --nproc_per_node=$GPU_NUM \
--train_batch_size=1 \
--max_height=480 \
--max_width=848 \
--target_length=163 \
--num_frames=163 \
--dataloader_num_workers 1 \
--output_dir=$OUTPUT_DIR