nicholasKluge commited on
Commit
a31e1ed
·
1 Parent(s): 1e1f4ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -33,9 +33,9 @@ The model was trained with a dataset composed of `demonstrations`, and annotated
33
  - **Learning Rate:** 1e-4
34
  - **Loss Function:** `torch.nn.MSELoss()`
35
  - **GPU:** 1 NVIDIA A100-SXM4-40GB
36
- - **RMSE in testing:** 0.1046
37
- - **Emissions:** 0.00 KgCO2
38
- - **Total Energy Consumption:** 0.00 kWh
39
 
40
 
41
  | Epoch/Loss|Training|Validation|
 
33
  - **Learning Rate:** 1e-4
34
  - **Loss Function:** `torch.nn.MSELoss()`
35
  - **GPU:** 1 NVIDIA A100-SXM4-40GB
36
+ - **RMSE in testing:** 0.1551
37
+ - **Emissions:** 0.38 KgCO2
38
+ - **Total Energy Consumption:** 0.85 kWh
39
 
40
 
41
  | Epoch/Loss|Training|Validation|