[feat]: Add adaptive fps dataloader and remove redundant code (#41)
No checkout mochi
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user