Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
23fb59e
1
Parent(s): b8e3e58
Update params.yml
Browse files- params.yml +1 -0
params.yml
CHANGED
|
@@ -6,6 +6,7 @@ model_name: t5-small
|
|
| 6 |
learning_rate: 1e-4
|
| 7 |
epochs: 5
|
| 8 |
source_dir: src
|
|
|
|
| 9 |
metric: rouge
|
| 10 |
split: 0.02
|
| 11 |
use_gpu: True
|
|
|
|
| 6 |
learning_rate: 1e-4
|
| 7 |
epochs: 5
|
| 8 |
source_dir: src
|
| 9 |
+
model_dir: models
|
| 10 |
metric: rouge
|
| 11 |
split: 0.02
|
| 12 |
use_gpu: True
|