update model card README.md
Browse files
README.md
CHANGED
|
@@ -36,7 +36,7 @@ on the `data/img_align_celeba` dataset.
|
|
| 36 |
### Training hyperparameters
|
| 37 |
|
| 38 |
The following hyperparameters were used during training:
|
| 39 |
-
- learning_rate:
|
| 40 |
- train_batch_size: 16
|
| 41 |
- eval_batch_size: 16
|
| 42 |
- gradient_accumulation_steps: 1
|
|
|
|
| 36 |
### Training hyperparameters
|
| 37 |
|
| 38 |
The following hyperparameters were used during training:
|
| 39 |
+
- learning_rate: 5e-06
|
| 40 |
- train_batch_size: 16
|
| 41 |
- eval_batch_size: 16
|
| 42 |
- gradient_accumulation_steps: 1
|