File size: 486 Bytes
533c844 |
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 |
log_dir: "checkpoint"
save_freq: 5
device: "cuda"
epochs: 30
batch_size: 32
pretrained_model: "/content/AuxiliaryASR/checkpoint/finetune.pth"
train_data: "/content/Data/train_asr.txt"
val_data: "/content/Data/train_asr.txt"
preprocess_parasm:
sr: 24000
spect_params:
n_fft: 2048
win_length: 1200
hop_length: 300
mel_params:
n_mels: 80
model_params:
input_dim: 80
hidden_dim: 256
n_token: 178
token_embedding_dim: 512
optimizer_params:
lr: 0.0005
|