high_fidelity_video_background_music_generation_with_transformers
/
models_peft
/symmv_small
/configuration.yml
log_level: INFO | |
num_epochs: 20 | |
batch_size: 16 | |
save_step: 200 | |
device: cuda | |
use_wandb: true | |
wandb_project_name: thesis_peft | |
gradient_clipping: 1.0 | |
beta1: 0.9 | |
beta2: 0.95 | |
weight_decay: 0.1 | |
learning_rate: 1.0e-05 | |
warmup_steps: 16 | |
video_extraction_framerate: 1 | |
encoder_input_dimension: 2048 | |
encoder_dim_feedforward: 8192 | |
encoder_heads: 16 | |
encoder_layers: 6 | |
musicgen_model_id: facebook/musicgen-stereo-small | |
use_peft: true | |
output_dir_name: symmv_small | |
dataset_video_folder: ../../datasets/symmv/videos_30sec | |
dataset_audio_folder: ../../datasets/symmv/audio_30sec_separated | |
use_demucs_folder_structure: true | |
dataset_validation_split: 0.1 | |
dataset_shuffling_seed: 42 | |