Update README.md
Browse files
README.md
CHANGED
|
@@ -34,6 +34,7 @@ More information needed
|
|
| 34 |
The following hyperparameters were used during training:
|
| 35 |
- learning_rate: 0.0001
|
| 36 |
- train_batch_size: 16
|
|
|
|
| 37 |
- eval_batch_size: 8
|
| 38 |
- seed: 42
|
| 39 |
- distributed_type: multi-GPU
|
|
|
|
| 34 |
The following hyperparameters were used during training:
|
| 35 |
- learning_rate: 0.0001
|
| 36 |
- train_batch_size: 16
|
| 37 |
+
- train_group_size: 4
|
| 38 |
- eval_batch_size: 8
|
| 39 |
- seed: 42
|
| 40 |
- distributed_type: multi-GPU
|