Update README.md
Browse files
README.md
CHANGED
@@ -90,7 +90,20 @@ The following hyperparameters were used during training:
|
|
90 |
- training precision: Mixed Precision
|
91 |
|
92 |
### Training results
|
93 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
94 |
|
95 |
|
96 |
### Framework versions
|
|
|
90 |
- training precision: Mixed Precision
|
91 |
|
92 |
### Training results
|
93 |
+
```
|
94 |
+
***** train metrics *****
|
95 |
+
"epoch": 4.0,
|
96 |
+
"train_loss": 0.6123406731570221,
|
97 |
+
"train_runtime": 29986.2288,
|
98 |
+
"train_samples": 943000,
|
99 |
+
"train_samples_per_second": 125.791,
|
100 |
+
"train_steps_per_second": 1.965
|
101 |
+
|
102 |
+
***** eval metrics *****
|
103 |
+
"eval_accuracy": 0.5933514030612245,
|
104 |
+
"eval_loss": 1.9326171875,
|
105 |
+
"eval_samples": 12576,
|
106 |
+
```
|
107 |
|
108 |
|
109 |
### Framework versions
|