|
log_level: INFO |
|
num_epochs: 20 |
|
batch_size: 2 |
|
save_step: 1800 |
|
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: 32768 |
|
encoder_heads: 64 |
|
encoder_layers: 24 |
|
musicgen_model_id: facebook/musicgen-stereo-large |
|
use_peft: false |
|
output_dir_name: audiocraft_nature_large |
|
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 |
|
|