File size: 1,792 Bytes
affcd23 |
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 95 96 97 98 99 100 101 |
{
"name": "test_config",
"n_gpu": 1,
"preprocessing": {
"sr": 16000,
"spectrogram": {
"type": "MelSpectrogram",
"args": {
}
},
"log_spec": true
},
"augmentations": {
"wave": [],
"spectrogram": []
},
"arch": {
"type": "BaselineModel",
"args": {
"n_feats": 128,
"fc_hidden": 512
}
},
"data": {
"train": {
"batch_size": 20,
"num_workers": 0,
"datasets": [
{
"type": "LibrispeechDataset",
"args": {
"part": "dev-clean",
"max_audio_length": 20.0,
"max_text_length": 200
}
}
]
},
"val": {
"batch_size": 20,
"num_workers": 0,
"datasets": [
{
"type": "LibrispeechDataset",
"args": {
"part": "dev-clean",
"max_audio_length": 20.0,
"max_text_length": 200
}
}
]
}
},
"optimizer": {
"type": "SGD",
"args": {
"lr": 3e-4
}
},
"loss": {
"type": "CTCLoss",
"args": {}
},
"metrics": [
{
"type": "ArgmaxWERMetric",
"args": {
"name": "WER (argmax)"
}
},
{
"type": "ArgmaxCERMetric",
"args": {
"name": "CER (argmax)"
}
}
],
"lr_scheduler": {
"type": "OneCycleLR",
"args": {
"steps_per_epoch": 100,
"epochs": 50,
"anneal_strategy": "cos",
"max_lr": 4e-3,
"pct_start": 0.2
}
},
"trainer": {
"epochs": 50,
"save_dir": "saved/",
"save_period": 5,
"verbosity": 2,
"monitor": "min val_loss",
"early_stop": 100,
"visualize": "wandb",
"wandb_project": "asr_project",
"len_epoch": 100,
"grad_norm_clip": 10
}
}
|