Spaces:
Runtime error
Runtime error
fixes
Browse files- params.yml +1 -1
params.yml
CHANGED
|
@@ -4,7 +4,7 @@ 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
|
| 10 |
metric: rouge
|
|
|
|
| 4 |
model_type: t5
|
| 5 |
model_name: t5-small
|
| 6 |
learning_rate: 1e-4
|
| 7 |
+
epochs: 3
|
| 8 |
source_dir: src
|
| 9 |
model_dir: models
|
| 10 |
metric: rouge
|