ffurfaro commited on
Commit
5678329
1 Parent(s): 4a670df

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -30,9 +30,9 @@ We use the **PixelBytes-Pokemon** dataset, available on Hugging Face: [PixelByte
30
 
31
  ## Models Trained
32
 
33
- - **8 LSTM Models**: Bidirectional + 1, 2, 3 layers (including p_embed + bi-2 layers)
34
- - **6 Mamba Models**: Bidirectional + 1, 2, 3 layers
35
- - **3 Transformer Models**: 1, 2, 3 layers
36
 
37
  ---
38
 
 
30
 
31
  ## Models Trained
32
 
33
+ - **10 LSTM Models**: (Uni-Bi)directional + 1, 2, 3 layers (including special config : p_embed + 3xhidden_state + 3xembedding_dim)
34
+ - **3 Mamba Models**: Bidirectional + 1, 2 layers, Unidirectional + 2 layers
35
+ - **2 Transformer Models**: 1, 2 layers
36
 
37
  ---
38