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

Co-authored-by: haoanyscale <c-hao.zhang@anyscale.com>
This commit is contained in:
rlsu9
2024-11-06 19:52:48 -08:00
committed by GitHub
co-authored by haoanyscale
parent 74688e56b5
commit 31827c9e0a
3 changed files with 16 additions and 114 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