| seed: 42 | |
| model: | |
| init_checkpoint: arda-argmax/diar_streaming_sortformer_6spk-init | |
| num_spks: 6 | |
| data: | |
| cuts_path: null | |
| dataset_hf_repo: null | |
| split: null | |
| cuts_filename: null | |
| preprocess: true | |
| audio_root: null | |
| session_len_sec: 90.0 | |
| sample_rate: 16000 | |
| batch_size: 1 | |
| num_workers: 0 | |
| shuffle: true | |
| streaming: | |
| enabled: true | |
| repo: arda-argmax/simchoir-parquet | |
| subset: debug | |
| train_split: train | |
| val_split: train | |
| train_shuffle_buffer: 0 | |
| sampler_shuffle_buffer: 0 | |
| optim: | |
| learning_rate: 5.0e-05 | |
| weight_decay: 0.0 | |
| warmup_steps: 50 | |
| min_lr: 1.0e-06 | |
| trainer: | |
| accelerator: gpu | |
| devices: 1 | |
| precision: 32 | |
| fallback: true | |
| num_steps: 3 | |
| val_check_interval: 1.0 | |
| check_val_every_n_epoch: 1 | |
| log_every_n_steps: 1 | |
| limit_val_batches: 1 | |
| num_sanity_val_steps: 0 | |
| regularization: | |
| disable_all_dropout: false | |
| causal_attn_rate: null | |
| scores_add_rnd: null | |
| output: | |
| final_checkpoint_save_path: ./.smoketest_final.nemo | |
| log_dir: ./.smoketest_logs | |
| run_name: smoke-1777401758 | |
| tracking: | |
| enabled: false | |
| project: sortformer-robustness | |
| run_name: null | |
| group: null | |
| space_id: null | |
| console_log: | |
| enabled: true | |
| max_chars: 100000 | |
| every_n_train_steps: 0 | |
| checkpointing: | |
| enabled: true | |
| repo: arda-argmax/sortformer-finetuned | |
| run_subdir: smoke-1777401758 | |
| monitor: val_loss | |
| mode: min | |
| private: true | |
| local_dir: null | |
| flush_timeout_sec: 120 | |