Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
library_name: keras
|
3 |
---
|
4 |
|
5 |
## Model description
|
@@ -38,4 +38,6 @@ The following hyperparameters were used during training:
|
|
38 |
| epsilon | 1e-07 |
|
39 |
| amsgrad | False |
|
40 |
| training_precision | float32 |
|
|
|
|
|
41 |
|
|
|
1 |
---
|
2 |
+
library_name: tf-keras
|
3 |
---
|
4 |
|
5 |
## Model description
|
|
|
38 |
| epsilon | 1e-07 |
|
39 |
| amsgrad | False |
|
40 |
| training_precision | float32 |
|
41 |
+
| epochs | 5 |
|
42 |
+
| batch size | 16 |
|
43 |
|