File size: 1,980 Bytes
07d5bbe
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
protocol: X.SpeakerDiarization.Custom
patience: 20
task:
  _target_: pyannote.audio.tasks.Segmentation
  duration: 5.0
  warm_up: 0.0
  balance: null
  overlap:
    probability: 0.5
    snr_min: 0.0
    snr_max: 10.0
  weight: null
  batch_size: 32
  num_workers: 10
  pin_memory: false
  loss: bce
  vad_loss: bce
model:
  _target_: pyannote.audio.models.segmentation.PyanNet
  sincnet:
    stride: 10
  lstm:
    num_layers: 4
    monolithic: true
    dropout: 0.5
  linear:
    num_layers: 2
optimizer:
  _target_: torch.optim.Adam
  lr: 0.001
  betas:
  - 0.9
  - 0.999
  eps: 1.0e-08
  weight_decay: 0
  amsgrad: false
trainer:
  _target_: pytorch_lightning.Trainer
  accelerator: ddp
  accumulate_grad_batches: 1
  amp_backend: native
  amp_level: O2
  auto_lr_find: false
  auto_scale_batch_size: false
  auto_select_gpus: true
  benchmark: true
  check_val_every_n_epoch: 1
  checkpoint_callback: true
  deterministic: false
  fast_dev_run: false
  flush_logs_every_n_steps: 100
  gpus: -1
  gradient_clip_val: 0.5
  limit_test_batches: 1.0
  limit_train_batches: 1.0
  limit_val_batches: 1.0
  log_every_n_steps: 50
  log_gpu_memory: null
  max_epochs: 1000
  max_steps: null
  min_epochs: 1
  min_steps: null
  num_nodes: 1
  num_processes: 1
  num_sanity_val_steps: 2
  overfit_batches: 0.0
  precision: 32
  prepare_data_per_node: true
  process_position: 0
  profiler: null
  progress_bar_refresh_rate: 1
  reload_dataloaders_every_epoch: false
  replace_sampler_ddp: true
  sync_batchnorm: false
  terminate_on_nan: false
  tpu_cores: null
  track_grad_norm: -1
  truncated_bptt_steps: null
  val_check_interval: 1.0
  weights_save_path: null
  weights_summary: top
augmentation:
  transform: Compose
  params:
    shuffle: false
    transforms:
    - transform: AddBackgroundNoise
      params:
        background_paths: /gpfswork/rech/eie/commun/data/background/musan
        min_snr_in_db: 5.0
        max_snr_in_db: 15.0
        mode: per_example
        p: 0.9