Update README.md
Browse files
README.md
CHANGED
|
@@ -47,10 +47,12 @@ model.load_state_dict(state_dict)
|
|
| 47 |
|
| 48 |
| Stat | Value |
|
| 49 |
| :------------------ | ----------: |
|
| 50 |
-
| #Parameters. |
|
| 51 |
-
| Best validation MAE | 0.
|
|
|
|
|
|
|
| 52 |
| Epochs trained | 1,000 |
|
| 53 |
-
| Best epoch. |
|
| 54 |
| Batch size | 256 |
|
| 55 |
| n_blocks | 6 |
|
| 56 |
| n_heads | 8 |
|
|
|
|
| 47 |
|
| 48 |
| Stat | Value |
|
| 49 |
| :------------------ | ----------: |
|
| 50 |
+
| #Parameters. | 111,555,513 |
|
| 51 |
+
| Best validation MAE | 0.0250 |
|
| 52 |
+
| top_p | 0.9 |
|
| 53 |
+
| top_k | 5 |
|
| 54 |
| Epochs trained | 1,000 |
|
| 55 |
+
| Best epoch. | 866 |
|
| 56 |
| Batch size | 256 |
|
| 57 |
| n_blocks | 6 |
|
| 58 |
| n_heads | 8 |
|