Update README.md
Browse files
README.md
CHANGED
|
@@ -221,6 +221,8 @@ register_conv_template(
|
|
| 221 |
|
| 222 |
### Training hyperparameters
|
| 223 |
|
|
|
|
|
|
|
| 224 |
The following hyperparameters were used during training:
|
| 225 |
- learning_rate: 1e-05
|
| 226 |
- train_batch_size: 2
|
|
|
|
| 221 |
|
| 222 |
### Training hyperparameters
|
| 223 |
|
| 224 |
+
This model was trained using 4 x A100 (80GB) for roughly 2.5 hours.
|
| 225 |
+
|
| 226 |
The following hyperparameters were used during training:
|
| 227 |
- learning_rate: 1e-05
|
| 228 |
- train_batch_size: 2
|