| audio: | |
| chunk_duration_s: 4 | |
| sample_rate: 16000 | |
| strict_frames: false | |
| auto_resume: false | |
| batch_size: '16' | |
| config: /lustre/fswork/projects/rech/kqg/commun/addressee_checkpoints_V3/checkpoints/bbh2Context10Sseed4/config.yml | |
| context_size: '10' | |
| data: | |
| classes: binary_classes | |
| dataset_path: /lustre/fswork/projects/rech/kqg/commun/addressee_dataset | |
| exp_config: pooling.yml | |
| freeze_encoder: false | |
| model_id: /lustre/fswork/projects/rech/kqg/commun/models/epoch=44-step=400000.ckpt | |
| model_type: hubert | |
| output: /lustre/fswork/projects/rech/kqg/commun/addressee_checkpoints_V3/checkpoints/ | |
| plots: false | |
| pool: true | |
| reduction: false | |
| run_id: bbh2Context10Sseed4 | |
| tags: [] | |
| train: | |
| batch_size: 16 | |
| classifier: 256 | |
| dataloader: | |
| num_workers: 8 | |
| extra_val_metrics: | |
| - loss | |
| - f1_score | |
| lr: 1.0e-05 | |
| mask_padding_attention: false | |
| max_epochs: 10 | |
| optim: | |
| decay_steps: 7500 | |
| final_lr_scale: 0.05 | |
| hold_steps: 12500 | |
| init_lr_scale: 0.01 | |
| scheduler: tristage | |
| warmup_steps: 5000 | |
| pad: true | |
| profiler: null | |
| scheduler: | |
| patience: 3 | |
| validation_metric: loss | |
| wandb: | |
| name: train | |
| offline: true | |
| project: addresseecontext | |