changed config to yaml!

1 files changed, 0 insertions, 17 deletions

diff --git a/swr2_asr/model_deep_speech.py b/swr2_asr/model_deep_speech.py
index 8ddbd99..77f4c8a 100644
--- a/swr2_asr/model_deep_speech.py
+++ b/swr2_asr/model_deep_speech.py
@@ -3,27 +3,10 @@
 Following definition by Assembly AI 
 (https://www.assemblyai.com/blog/end-to-end-speech-recognition-pytorch/)
 """
-from typing import TypedDict
-
 import torch.nn.functional as F
 from torch import nn
 
 
-class HParams(TypedDict):
-    """Type for the hyperparameters of the model."""
-
-    n_cnn_layers: int
-    n_rnn_layers: int
-    rnn_dim: int
-    n_class: int
-    n_feats: int
-    stride: int
-    dropout: float
-    learning_rate: float
-    batch_size: int
-    epochs: int
-
-
 class CNNLayerNorm(nn.Module):
     """Layer normalization built for cnns input"""