Commit
·
a31e1ed
1
Parent(s):
1e1f4ef
Update README.md
Browse files
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.
|
37 |
-
- **Emissions:** 0.
|
38 |
-
- **Total Energy Consumption:** 0.
|
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|
|