From 58b30927bd870604a4077a8af9ec3cad7b0be21c Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Mon, 11 Sep 2023 21:52:42 +0200
Subject: changed config to yaml!

---
 swr2_asr/model_deep_speech.py | 17 -----------------
 1 file changed, 17 deletions(-)

(limited to 'swr2_asr/model_deep_speech.py')

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"""
 
-- 
cgit v1.2.3