brendenc commited on
Commit
7cf45a1
1 Parent(s): 347a607

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -26,6 +26,8 @@ The training was done using TensorFlow 2.8.0 and TensorFlow Probability 0.16.0.
26
 
27
  ### Training hyperparameters
28
 
29
- The following hyperparameters were used during training:
30
- - optimizer: {'name': 'RMSprop', 'learning_rate': 0.001, 'decay': 0.0, 'rho': 0.9, 'momentum': 0.0, 'epsilon': 1e-07, 'centered': False}
31
- - training_precision: float32
 
 
26
 
27
  ### Training hyperparameters
28
 
29
+ | Optimizer | learning_rate | decay | rho | momentum | epsilon | centered | training_precision |
30
+ |----|-------------|-----|------|------|-------|-------|------------------|
31
+ |RMSprop|0.001|0.0|0.9|0.0|1e-07|False|float32|
32
+
33
+