Commit
·
d291863
1
Parent(s):
ab2d8a2
model files
Browse files- config.yaml +91 -0
- pytorch_model.ckpt +3 -0
config.yaml
ADDED
|
@@ -0,0 +1,91 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model:
|
| 2 |
+
_target_: enhancer.models.waveunet.WaveUnet
|
| 3 |
+
num_channels: 1
|
| 4 |
+
depth: 9
|
| 5 |
+
initial_output_channels: 24
|
| 6 |
+
sampling_rate: 16000
|
| 7 |
+
dataset:
|
| 8 |
+
_target_: enhancer.data.dataset.EnhancerDataset
|
| 9 |
+
name: vctk
|
| 10 |
+
root_dir: /scratch/c.sistc3/DS_10283_2791
|
| 11 |
+
duration: 2
|
| 12 |
+
stride: 1
|
| 13 |
+
sampling_rate: 16000
|
| 14 |
+
batch_size: 128
|
| 15 |
+
valid_minutes: 25
|
| 16 |
+
files:
|
| 17 |
+
train_clean: clean_trainset_28spk_wav
|
| 18 |
+
test_clean: clean_testset_wav
|
| 19 |
+
train_noisy: noisy_trainset_28spk_wav
|
| 20 |
+
test_noisy: noisy_testset_wav
|
| 21 |
+
optimizer:
|
| 22 |
+
_target_: torch.optim.Adam
|
| 23 |
+
lr: 0.001
|
| 24 |
+
betas:
|
| 25 |
+
- 0.9
|
| 26 |
+
- 0.999
|
| 27 |
+
eps: 1.0e-08
|
| 28 |
+
weight_decay: 0
|
| 29 |
+
amsgrad: false
|
| 30 |
+
hyperparameters:
|
| 31 |
+
loss: mae
|
| 32 |
+
metric:
|
| 33 |
+
- stoi
|
| 34 |
+
- pesq
|
| 35 |
+
- si-sdr
|
| 36 |
+
lr: 0.003
|
| 37 |
+
ReduceLr_patience: 10
|
| 38 |
+
Early_stop: false
|
| 39 |
+
ReduceLr_factor: 0.1
|
| 40 |
+
min_lr: 1.0e-07
|
| 41 |
+
EarlyStopping_patience: 10
|
| 42 |
+
trainer:
|
| 43 |
+
_target_: pytorch_lightning.Trainer
|
| 44 |
+
accelerator: gpu
|
| 45 |
+
accumulate_grad_batches: 1
|
| 46 |
+
amp_backend: native
|
| 47 |
+
auto_lr_find: false
|
| 48 |
+
auto_scale_batch_size: false
|
| 49 |
+
auto_select_gpus: true
|
| 50 |
+
benchmark: false
|
| 51 |
+
check_val_every_n_epoch: 1
|
| 52 |
+
detect_anomaly: false
|
| 53 |
+
deterministic: false
|
| 54 |
+
devices: 1
|
| 55 |
+
enable_checkpointing: true
|
| 56 |
+
enable_model_summary: true
|
| 57 |
+
enable_progress_bar: true
|
| 58 |
+
fast_dev_run: false
|
| 59 |
+
gpus: null
|
| 60 |
+
gradient_clip_val: 0
|
| 61 |
+
gradient_clip_algorithm: norm
|
| 62 |
+
ipus: null
|
| 63 |
+
limit_predict_batches: 1.0
|
| 64 |
+
limit_test_batches: 1.0
|
| 65 |
+
limit_train_batches: 1.0
|
| 66 |
+
limit_val_batches: 1.0
|
| 67 |
+
log_every_n_steps: 50
|
| 68 |
+
max_epochs: 200
|
| 69 |
+
max_steps: -1
|
| 70 |
+
max_time: null
|
| 71 |
+
min_epochs: 1
|
| 72 |
+
min_steps: null
|
| 73 |
+
move_metrics_to_cpu: false
|
| 74 |
+
multiple_trainloader_mode: max_size_cycle
|
| 75 |
+
num_nodes: 1
|
| 76 |
+
num_processes: 1
|
| 77 |
+
num_sanity_val_steps: 2
|
| 78 |
+
overfit_batches: 0.0
|
| 79 |
+
precision: 32
|
| 80 |
+
profiler: null
|
| 81 |
+
reload_dataloaders_every_n_epochs: 0
|
| 82 |
+
replace_sampler_ddp: true
|
| 83 |
+
strategy: null
|
| 84 |
+
sync_batchnorm: false
|
| 85 |
+
tpu_cores: null
|
| 86 |
+
track_grad_norm: -1
|
| 87 |
+
val_check_interval: 1.0
|
| 88 |
+
weights_save_path: null
|
| 89 |
+
mlflow:
|
| 90 |
+
experiment_name: shahules/enhancer
|
| 91 |
+
run_name: WaveUnet-vctk
|
pytorch_model.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d6554d04212225b4ec99077ed2dda13436fbe9fa33a55d6a331feea7c7e856d6
|
| 3 |
+
size 54447197
|