Spaces:
Runtime error
Runtime error
fixes
Browse files- params.yml +3 -3
params.yml
CHANGED
|
@@ -4,9 +4,9 @@ num_workers: 2
|
|
| 4 |
model_type: t5
|
| 5 |
model_name: t5-small
|
| 6 |
learning_rate: 1e-4
|
| 7 |
-
epochs:
|
| 8 |
source_dir: src
|
| 9 |
-
model_dir: models/models
|
| 10 |
metric: rouge
|
| 11 |
-
split: 0.
|
| 12 |
use_gpu: True
|
|
|
|
| 4 |
model_type: t5
|
| 5 |
model_name: t5-small
|
| 6 |
learning_rate: 1e-4
|
| 7 |
+
epochs: 5
|
| 8 |
source_dir: src
|
| 9 |
+
model_dir: models/models/
|
| 10 |
metric: rouge
|
| 11 |
+
split: 0.02
|
| 12 |
use_gpu: True
|