| { |
| "model": "speaker_encoder", |
| "run_name": "speaker_encoder", |
| "run_description": "resnet speaker encoder trained with commonvoice all languages dev and train, Voxceleb 1 dev and Voxceleb 2 dev", |
| "epochs": 100000, |
| "batch_size": null, |
| "eval_batch_size": null, |
| "mixed_precision": false, |
| "run_eval": true, |
| "test_delay_epochs": 0, |
| "print_eval": false, |
| "print_step": 50, |
| "tb_plot_step": 100, |
| "tb_model_param_stats": false, |
| "save_step": 1000, |
| "checkpoint": true, |
| "keep_all_best": false, |
| "keep_after": 10000, |
| "num_loader_workers": 8, |
| "num_val_loader_workers": 0, |
| "use_noise_augment": false, |
| "output_path": "../checkpoints/speaker_encoder/language_balanced/normalized/angleproto-4-samples-by-speakers/", |
| "distributed_backend": "nccl", |
| "distributed_url": "tcp://localhost:54321", |
| "audio": { |
| "fft_size": 512, |
| "win_length": 400, |
| "hop_length": 160, |
| "frame_shift_ms": null, |
| "frame_length_ms": null, |
| "stft_pad_mode": "reflect", |
| "sample_rate": 16000, |
| "resample": false, |
| "preemphasis": 0.97, |
| "ref_level_db": 20, |
| "do_sound_norm": false, |
| "do_trim_silence": false, |
| "trim_db": 60, |
| "power": 1.5, |
| "griffin_lim_iters": 60, |
| "num_mels": 64, |
| "mel_fmin": 0.0, |
| "mel_fmax": 8000.0, |
| "spec_gain": 20, |
| "signal_norm": false, |
| "min_level_db": -100, |
| "symmetric_norm": false, |
| "max_norm": 4.0, |
| "clip_norm": false, |
| "stats_path": null |
| }, |
| "datasets": [ |
| { |
| "name": "voxceleb2", |
| "path": "/workspace/scratch/ecasanova/datasets/VoxCeleb/vox2_dev_aac/", |
| "meta_file_train": null, |
| "ununsed_speakers": null, |
| "meta_file_val": null, |
| "meta_file_attn_mask": "", |
| "language": "voxceleb" |
| } |
| ], |
| "model_params": { |
| "model_name": "resnet", |
| "input_dim": 64, |
| "use_torch_spec": true, |
| "log_input": true, |
| "proj_dim": 512 |
| }, |
| "audio_augmentation": { |
| "p": 0.5, |
| "rir": { |
| "rir_path": "/workspace/store/ecasanova/ComParE/RIRS_NOISES/simulated_rirs/", |
| "conv_mode": "full" |
| }, |
| "additive": { |
| "sounds_path": "/workspace/store/ecasanova/ComParE/musan/", |
| "speech": { |
| "min_snr_in_db": 13, |
| "max_snr_in_db": 20, |
| "min_num_noises": 1, |
| "max_num_noises": 1 |
| }, |
| "noise": { |
| "min_snr_in_db": 0, |
| "max_snr_in_db": 15, |
| "min_num_noises": 1, |
| "max_num_noises": 1 |
| }, |
| "music": { |
| "min_snr_in_db": 5, |
| "max_snr_in_db": 15, |
| "min_num_noises": 1, |
| "max_num_noises": 1 |
| } |
| }, |
| "gaussian": { |
| "p": 0.0, |
| "min_amplitude": 0.0, |
| "max_amplitude": 1e-05 |
| } |
| }, |
| "storage": { |
| "sample_from_storage_p": 0.5, |
| "storage_size": 40 |
| }, |
| "max_train_step": 1000000, |
| "loss": "angleproto", |
| "grad_clip": 3.0, |
| "lr": 0.0001, |
| "lr_decay": false, |
| "warmup_steps": 4000, |
| "wd": 1e-06, |
| "steps_plot_stats": 100, |
| "num_speakers_in_batch": 100, |
| "num_utters_per_speaker": 4, |
| "skip_speakers": true, |
| "voice_len": 2.0 |
| } |
