Update README.md
Browse files
README.md
CHANGED
@@ -33,7 +33,12 @@ Mariana Moreira dos Santos ([LinkedIn](https://www.linkedin.com/in/mmoreirast/))
|
|
33 |
- **GPU:**
|
34 |
- **Training time**: ~
|
35 |
- **Emissions:**
|
36 |
-
- **Total energy consumption:**
|
|
|
|
|
|
|
|
|
|
|
37 |
|
38 |
## Code
|
39 |
This repository has the *GOOGLE COLAB LINK* used to train this model.
|
|
|
33 |
- **GPU:**
|
34 |
- **Training time**: ~
|
35 |
- **Emissions:**
|
36 |
+
- **Total energy consumption:**
|
37 |
+
|
38 |
+
## Parameters
|
39 |
+
- **Number of Epochs:** 3
|
40 |
+
- **Batch size:** 4
|
41 |
+
- **Optimizer:** torch.optim.AdamW (warmup_steps = 1e3, learning_rate = 1e-5, epsilon = 1e-8)
|
42 |
|
43 |
## Code
|
44 |
This repository has the *GOOGLE COLAB LINK* used to train this model.
|