Hervé BREDIN commited on
Commit
3a5e33c
1 Parent(s): dd586f4

initial import

Browse files
Files changed (3) hide show
  1. README.md +16 -0
  2. config.yaml +61 -0
  3. pytorch_model.bin +3 -0
README.md CHANGED
@@ -1,3 +1,19 @@
1
  ---
 
 
 
 
2
  license: mit
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ tags:
3
+ - pyannote
4
+ - pyannote-audio
5
+ - pyannote-audio-model
6
  license: mit
7
+ inference: false
8
  ---
9
+
10
+ Dummy segmentation model used for continuous integration and unit tests.
11
+
12
+ ```bash
13
+ pyannote-audio-train \
14
+ +registry="[REDACTED]/pyannote-audio/tutorials/AMI-diarization-setup/pyannote/database.yml" \
15
+ protocol=AMI.SpeakerDiarization.only_words \
16
+ model=DebugSegmentation \
17
+ task=SpeakerDiarization \
18
+ trainer.max_epochs=1
19
+ ```
config.yaml ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ task:
2
+ _target_: pyannote.audio.tasks.SpeakerDiarization
3
+ duration: 5.0
4
+ max_speakers_per_chunk: 3
5
+ max_speakers_per_frame: 2
6
+ batch_size: 32
7
+ num_workers: 10
8
+ pin_memory: false
9
+ model:
10
+ _target_: pyannote.audio.models.segmentation.debug.SimpleSegmentationModel
11
+ optimizer:
12
+ _target_: torch.optim.Adam
13
+ lr: 0.001
14
+ betas:
15
+ - 0.9
16
+ - 0.999
17
+ eps: 1.0e-08
18
+ weight_decay: 0
19
+ amsgrad: false
20
+ scheduler:
21
+ _target_: pyannote.audio.cli.lr_schedulers.CosineAnnealingWarmRestarts
22
+ min_lr: 1.0e-08
23
+ max_lr: 0.001
24
+ patience: 1
25
+ trainer:
26
+ _target_: pytorch_lightning.Trainer
27
+ accelerator: auto
28
+ accumulate_grad_batches: 1
29
+ benchmark: null
30
+ deterministic: false
31
+ check_val_every_n_epoch: 1
32
+ devices: auto
33
+ detect_anomaly: false
34
+ enable_checkpointing: true
35
+ enable_model_summary: true
36
+ enable_progress_bar: true
37
+ fast_dev_run: false
38
+ gradient_clip_val: null
39
+ gradient_clip_algorithm: norm
40
+ limit_predict_batches: 1.0
41
+ limit_test_batches: 1.0
42
+ limit_train_batches: 1.0
43
+ limit_val_batches: 1.0
44
+ log_every_n_steps: 50
45
+ max_epochs: 1
46
+ max_steps: -1
47
+ max_time: null
48
+ min_epochs: 1
49
+ min_steps: null
50
+ num_nodes: 1
51
+ num_sanity_val_steps: 2
52
+ overfit_batches: 0.0
53
+ precision: 32
54
+ profiler: null
55
+ reload_dataloaders_every_n_epochs: 0
56
+ use_distributed_sampler: true
57
+ strategy: auto
58
+ sync_batchnorm: false
59
+ val_check_interval: 1.0
60
+ protocol: AMI.SpeakerDiarization.only_words
61
+ registry: REDACTED/pyannote-audio/tutorials/AMI-diarization-setup/pyannote/database.yml
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7656a3ba0cd193c2f9eb17fe2e5925c1eb680bc4d63fe4a2c92eeb2c574a7d2b
3
+ size 375564