seed: 42
force_exit_on_completion: true
model:
  init_checkpoint: arda-argmax/diar_streaming_sortformer_6spk-v0.6
  num_spks: 6
data:
  cuts_path: null
  dataset_hf_repo: null
  split: null
  cuts_filename: null
  preprocess: true
  audio_root: null
  session_len_sec: 120
  sample_rate: 16000
  batch_size: 10
  num_workers: 4
  shuffle: true
  prepared:
    enabled: true
    uri: gs://fastmss-synth/sortformer/fastmss_synth/vox-ami-aishell-ali
    local_dir: /root/vox_ami_aishell_ali_shar
    root_dir: null
    train_dir: null
    val_dir: null
    train_path: train
    val_path: val
    sync: true
    gcloud_bin: gcloud
    max_train_cuts: null
    max_val_cuts: null
    num_train_samples: 8400
  streaming:
    enabled: false
    repo: arda-argmax/simchoir-parquet
    subset: v0.1
    label_source: legacy
    cache_materialization: stream
    download_dir: /tmp/sortformer_hf_parquet_chunks
    materialization_workers: 1
    subset_prefix: null
    train_subsets: null
    val_subsets: null
    train_split: train
    val_split: val
    train_shuffle_buffer: 0
    sampler_shuffle_buffer: 1000
    num_train_samples: 800
    train_cache_size: -1
    cache_storage: disk
    cache_dir: /tmp/sortformer_streaming_cache
    val_cache_size: -1
    val_replicate_across_ranks: false
    bucketing:
      enabled: false
      max_duration_sec: null
      num_buckets: 10
      buffer_size: 256
      quadratic_duration: 30.0
      duration_bins: []
      bin_estimate_samples: 10000
optim:
  learning_rate: 3.0e-05
  weight_decay: 0.001
  warmup_steps: 26
  min_lr: 1.0e-06
trainer:
  accelerator: gpu
  devices: auto
  precision: bf16-mixed
  fallback: false
  num_steps: -1
  num_epochs: 5
  val_check_interval: 0.25
  check_val_every_n_epoch: 1
  log_every_n_steps: 1
  limit_val_batches: 1.0
  num_sanity_val_steps: 0
  accumulate_grad_batches: 2
  strategy: auto
regularization:
  disable_all_dropout: false
  causal_attn_rate: null
  scores_add_rnd: null
output:
  final_checkpoint_save_path: /app/training_runs/1782376025/final.nemo
  log_dir: /app/training_runs/1782376025/logs
  run_name: 6spk-vox-ami-aishell-ali-bs10-ga2-a100-80gb-4gpu-5ep-v1-1782376025
tracking:
  enabled: true
  project: sortformer-robustness
  run_name: 6spk-vox-ami-aishell-ali-bs10-ga2-a100-80gb-4gpu-5ep-v1-1782376025
  group: null
  space_id: null
  auto_log_gpu: true
  gpu_log_interval: 10.0
  console_log:
    enabled: false
    max_chars: 200000
    every_n_train_steps: 0
val_der:
  enabled: true
  collar: 0.0
  skip_overlap: false
  threshold: 0.5
  max_samples: null
optim_metrics:
  enabled: true
  every_n_steps: 50
  param_norm: true
  grad_norm: true
  weight_decay: true
checkpointing:
  enabled: true
  repo: arda-argmax/sortformer-finetuned
  run_subdir: 6spk-vox-ami-aishell-ali-bs10-ga2-a100-80gb-4gpu-5ep-v1-1782376025
  monitor: val_der
  mode: min
  private: true
  local_dir: null
  flush_timeout_sec: 1800