| log_level: INFO | |
| num_epochs: 20 | |
| batch_size: 8 | |
| save_step: 350 | |
| device: cuda | |
| use_wandb: true | |
| wandb_project_name: thesis_audiocraft | |
| 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: 16384 | |
| encoder_heads: 32 | |
| encoder_layers: 12 | |
| musicgen_model_id: facebook/musicgen-stereo-small | |
| use_peft: false | |
| output_dir_name: audiocraft_nature_small | |
| dataset_video_folder: ../../datasets/nature2/videos_30sec | |
| dataset_audio_folder: ../../datasets/nature2/audio_30sec | |
| use_demucs_folder_structure: false | |
| dataset_validation_split: 0.1 | |
| dataset_shuffling_seed: 42 | |