Spaces:
Runtime error
Runtime error
Update model_params.yml
Browse files- model_params.yml +7 -7
model_params.yml
CHANGED
|
@@ -1,11 +1,11 @@
|
|
| 1 |
-
name: summarsiation
|
| 2 |
-
model_type: t5
|
| 3 |
-
model_name: t5-small
|
| 4 |
batch_size: 2
|
| 5 |
epochs: 5
|
| 6 |
-
|
| 7 |
-
learning_rate: 1e-4
|
| 8 |
-
num_workers: 2
|
| 9 |
-
model_dir: models
|
| 10 |
metric: rouge
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
source_dir: src
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
batch_size: 2
|
| 2 |
epochs: 5
|
| 3 |
+
learning_rate: 0.0001
|
|
|
|
|
|
|
|
|
|
| 4 |
metric: rouge
|
| 5 |
+
model_dir: models
|
| 6 |
+
model_name: t5-base
|
| 7 |
+
model_type: t5
|
| 8 |
+
name: summarsiation
|
| 9 |
+
num_workers: 2
|
| 10 |
source_dir: src
|
| 11 |
+
use_gpu: true
|