pfialho commited on
Commit
a6bbb73
1 Parent(s): edc64c4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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