Upload config.json

config.json

@@ -0,0 +1,98 @@
+{
+    "model_name": "Efficient Conformer CTC Small",
+    "model_type": "CTC",
+
+    "encoder_params": 
+    {
+        "arch": "Conformer",
+        "num_blocks": 15,
+        "dim_model": [120, 168, 240],
+        "ff_ratio": 4,
+        "num_heads": 4,
+        "kernel_size": 15,
+        "Pdrop": 0.1,
+        "conv_stride": 2,
+        "att_stride": 1,
+        "strided_blocks": [4, 9],
+        "expand_blocks": [4, 9],
+        "att_group_size": [3, 1, 1],
+
+        "relative_pos_enc": true,
+        "max_pos_encoding": 10000,
+
+        "subsampling_module": "Conv2d",
+        "subsampling_layers": 1,
+        "subsampling_filters": [120],
+        "subsampling_kernel_size": 3,
+        "subsampling_norm": "batch",
+        "subsampling_act": "swish",
+
+        "sample_rate": 16000,
+        "win_length_ms": 25,
+        "hop_length_ms": 10,
+        "n_fft": 512,
+        "n_mels": 80,
+        "normalize": false,
+        "mean": -5.6501,
+        "std": 4.2280,
+
+        "spec_augment": false,
+        "mF": 2,
+        "F": 27,
+        "mT": 5,
+        "pS": 0.05
+    },
+    
+    "tokenizer_params":
+    {
+        "tokenizer_path": "datasets/Vietnamese/vi_bpe_1024.model",
+        "vocab_type": "bpe",
+        "vocab_size": 1024
+    },
+
+    "training_params":
+    {
+        "epochs": 450,
+        "batch_size": 64,
+        "accumulated_steps": 2,
+        "mixed_precision": true,
+
+        "optimizer": "Adam",
+        "beta1": 0.9,
+        "beta2": 0.98,
+        "eps": 1e-9,
+        "weight_decay": 1e-6,
+
+        "lr_schedule": "Transformer",
+        "schedule_dim": 240,
+        "warmup_steps": 10000,
+        "K": 2,
+
+        "train_audio_max_length": 256000,
+        "train_label_max_length": 256000,
+        "eval_audio_max_length": 256000,
+        "eval_label_max_length": 256000,
+
+        "training_dataset": "Vietnamese",
+        "training_dataset_path": "/mnt/c/Users/hyngu/Data/ASRDataset/",
+
+        "evaluation_dataset": "Vietnamese",
+        "evaluation_dataset_path": "/mnt/c/Users/hyngu/Data/ASRDataset/",
+
+        "callback_path": "callbacks/EfficientConformerCTCSmall/"
+    },
+
+    "decoding_params":
+    {
+        "beam_size": 5,
+        "tmp": 1,
+
+        "ngram_path": "data/6gram_lm_corpus.binary",
+        "ngram_alpha": 0.4,
+        "ngram_beta": 1.0,
+
+        "lm_config": "configs/LM-Transformer.json",
+        "lm_weight": 1,
+        "lm_tmp": 1
+    }
+}