checkpoint: exp/redimnet2/expB6_v2/-LM/models/model_0.pt
data_type: shard
dataloader_args:
  batch_size: 16
  drop_last: true
  num_workers: 4
  pin_memory: false
  prefetch_factor: 4
dataset_args:
  aug_prob: 0.0
  cmvn: false
  filter: false
  frontend: tfmel
  num_frms: 600
  resample_rate: 16000
  sample_num_per_epoch: 0
  shuffle: true
  shuffle_args:
    shuffle_size: 2500
  spec_aug: false
  speed_perturb: false
  tfmel_args:
    do_preemph: true
    do_spec_aug: false
    eps: 1.0e-08
    f_max: 7600
    f_min: 20
    hop_length: 160
    n_fft: 512
    n_mels: 72
    norm_signal: true
    win_length: 400
enable_amp: true
exp_dir: exp/redimnet2/expB6_v2/-LM
gpus:
- 2
- 3
grad_accum_steps: 8
log_batch_interval: 100
loss: CrossEntropyLoss
loss_args: {}
margin_scheduler: MarginScheduler
margin_update:
  epoch_iter: 34125
  final_margin: 0.3
  fix_start_epoch: 0
  increase_start_epoch: 0
  increase_type: exp
  initial_margin: 0.3
  update_margin: false
model: ReDimNet2Wrap
model_args:
  C: 64
  F: 72
  block_1d_type: conv+att
  block_2d_type: basic_resnet
  causal: none
  compress_tconvs: true
  emb_bn: false
  embed_dim: 192
  feat_dim: 72
  fm_weigthing_type: NC
  global_context_att: true
  group_divisor: 1
  hop_length: 160
  out_channels: 224
  pooling_func: ASTP
  return_2d_output: true
  spec: null
  spec_in_channels: 1
  stages_setup:
  - - - 1
      - 1
    - 3
    - 3
    - - - 3
        - 3
    - 64
  - - - 2
      - 1
    - 4
    - 2
    - - - 3
        - 3
    - 64
  - - - 1
      - 2
    - 5
    - 2
    - - - 3
        - 3
    - 48
  - - - 2
      - 1
    - 5
    - 1
    - - - 3
        - 3
    - 48
  - - - 1
      - 2
    - 4
    - 0.75
    - - - 3
        - 3
    - 32
  - - - 2
      - 1
    - 3
    - 0.5
    - - - 3
        - 3
    - 24
model_init: null
noise_data: data/voxceleb/musan/lmdb
num_avg: 1
num_epochs: 5
optimizer: SGD
optimizer_args:
  lr: 0.0001
  momentum: 0.9
  nesterov: true
  weight_decay: 2.0e-05
projection_args:
  do_lm: false
  easy_margin: false
  embed_dim: 192
  lanbuda: 0.7
  margin_type: C
  num_class: 5994
  project_type: sphereface2
  scale: 32.0
  t: 3
reverb_data: data/voxceleb/rirs/lmdb
save_epoch_interval: 1
scheduler: ExponentialDecrease
scheduler_args:
  epoch_iter: 34125
  final_lr: 1.0e-05
  initial_lr: 0.0001
  num_epochs: 5
  scale_ratio: 4.0
  warm_from_zero: false
  warm_up_epoch: 0
seed: 42
train_data: data/voxceleb/vox2_dev/shard.list
train_label: data/voxceleb/vox2_dev/utt2spk