hparams.py

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,  
    }