File size: 658 Bytes
0e8f8bc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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
|