try / config.yaml
zhuhz22's picture
End of training
13b34f8 verified
raw
history blame
989 Bytes
use_ema: true
output_dir: results/train
pretrained_model_path: /mnt/vepfs/zhaomin/projects/LargeData/checkpoints/stable-video-diffusion-img2vid/
train_data:
file_path: /mnt/vepfs/zhaomin/projects/scence_fusion/improved-svd/finalre/webvid.csv
video_folder: /mnt/vepfs/bigdata/video_datasets/webvid/2M_train
sample_size: 320,512
fps: 3
sample_n_frames: 16
validation_folder: /mnt/vepfs/zhaomin/projects/scence_fusion/improved-svd/SVD_diffusers/demo
motion_bucket_id: 20.0
cfg_random_null_ratio: 0.1
resume_path: ''
ema_decay: 0.9999
noise_scheduler_kwargs:
P_mean: -1.2
P_std: 1.2
sigma_data: 1
beta_m: 15
a: 5
max_train_steps: 100000
validation_steps: 10
learning_rate: 3.0e-05
scale_lr: false
lr_scheduler: constant
train_batch_size: 1
adam_beta1: 0.9
adam_beta2: 0.999
adam_weight_decay: 0.01
adam_epsilon: 1.0e-08
gradient_accumulation_steps: 1
gradient_checkpointing: true
checkpointing_steps: 10000
mixed_precision_training: true
global_seed: 23
max_grad_norm: 1.0