Ivanhoe9 commited on
Commit
6f9ec7b
·
verified ·
1 Parent(s): a92ff78

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -88,9 +88,9 @@ The model was trained on a dataset of texts and their respective titles, preproc
88
 
89
  #### Training Hyperparameters
90
 
91
- - **Batch size:** 6
92
- - **Number of epochs:** 15
93
- - **Learning rate:** 1e-5
94
  - **Optimizer:** Adam
95
  - **Loss function:** Cross-Entropy Loss
96
 
@@ -117,9 +117,9 @@ Evaluation was performed using a portion of the dataset held out during training
117
 
118
  The carbon footprint and energy consumption can be estimated using tools like the [Machine Learning Impact Calculator](https://mlco2.github.io/impact#compute).
119
 
120
- - **Hardware Type:** GPU (NVIDIA)
121
- - **Hours used:**
122
- - **Compute Region:**
123
 
124
  ## Technical Specifications
125
 
 
88
 
89
  #### Training Hyperparameters
90
 
91
+ - **Batch size:** 32
92
+ - **Number of epochs:** 9
93
+ - **Learning rate:** 1e-4
94
  - **Optimizer:** Adam
95
  - **Loss function:** Cross-Entropy Loss
96
 
 
117
 
118
  The carbon footprint and energy consumption can be estimated using tools like the [Machine Learning Impact Calculator](https://mlco2.github.io/impact#compute).
119
 
120
+ - **Hardware Type:** GPU A100 (NVIDIA)
121
+ - **Hours used:** 0.5 H
122
+ - **Compute Region:** [GLICID HPC](https://www.glicid.fr/)
123
 
124
  ## Technical Specifications
125