| { | |
| "specaug": { | |
| "freq_mask_width_range": 0.35, | |
| "num_freq_mask": 2, | |
| "num_time_mask": 10, | |
| "time_mask_width_range": 0.05 | |
| }, | |
| "encoder": { | |
| "type": "ConformerNet", | |
| "kwargs": { | |
| "num_cells": 22, | |
| "idim": 80, | |
| "hdim": 640, | |
| "conv": "vgg2l", | |
| "num_heads": 4, | |
| "d_head": 160, | |
| "kernel_size": 15, | |
| "num_classes": 75 | |
| } | |
| }, | |
| "scheduler": { | |
| "type": "SchedulerNoamEarlyStop", | |
| "kwargs": { | |
| "warmup_step": 20000, | |
| "min_step": 30000, | |
| "dim_model": 640, | |
| "peak_factor": 1.0, | |
| "n_tol": 50 | |
| }, | |
| "optimizer": { | |
| "type": "Adam", | |
| "kwargs": { | |
| "lr": 0.0003, | |
| "betas": [ | |
| 0.9, | |
| 0.98 | |
| ], | |
| "weight_decay": 1e-06 | |
| } | |
| } | |
| } | |
| } | |