mmoreirast commited on
Commit
86fc256
·
verified ·
1 Parent(s): 1729db4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
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.