Spaces:
Paused
Paused
Update src/config.yaml
Browse files- src/config.yaml +1 -1
src/config.yaml
CHANGED
|
@@ -18,7 +18,7 @@ params:
|
|
| 18 |
lr: 1e-4
|
| 19 |
warmup_ratio: 0.1
|
| 20 |
weight_decay: 0.01
|
| 21 |
-
epochs:
|
| 22 |
batch_size: 2
|
| 23 |
gradient_accumulation: 8
|
| 24 |
mixed_precision: fp16
|
|
|
|
| 18 |
lr: 1e-4
|
| 19 |
warmup_ratio: 0.1
|
| 20 |
weight_decay: 0.01
|
| 21 |
+
epochs: 1
|
| 22 |
batch_size: 2
|
| 23 |
gradient_accumulation: 8
|
| 24 |
mixed_precision: fp16
|