| import os | |
| import sys | |
| import torch | |
| class Hparams: | |
| args = { | |
| 'save_model_dir': '/kaggle/output/results/lr1e-3', | |
| 'device': 'cuda' if torch.cuda.is_available() else 'cpu', | |
| 'dataset_root': '/kaggle/input/paddle-speech/', | |
| 'sampling_rate': 16000, | |
| 'sample_length': 16, # in second | |
| 'num_workers': 4, # Number of additional thread for data loading. A large number may freeze your laptop. | |
| 'frame_size': 0.025, | |
| 'batch_size': 32, | |
| } | |