Update README.md
Browse files
README.md
CHANGED
@@ -8,7 +8,9 @@ tags:
|
|
8 |
datasets: pt-sk/research_papers_short
|
9 |
metrics: CrossEntropyLoss
|
10 |
---
|
11 |
-
This model using Mamba Architecuture trained on a research abstract dataset.
|
|
|
|
|
12 |
|
13 |
* Optimizer: AdamW
|
14 |
* Leanring Rate: 0.001
|
|
|
8 |
datasets: pt-sk/research_papers_short
|
9 |
metrics: CrossEntropyLoss
|
10 |
---
|
11 |
+
This model using Mamba Architecuture trained on a research abstract dataset.
|
12 |
+
|
13 |
+
mamba_reserach file contains the state dict of optimizer and the model.
|
14 |
|
15 |
* Optimizer: AdamW
|
16 |
* Leanring Rate: 0.001
|