Spaces:
Runtime error
Runtime error
fixes
Browse files- params.yml +2 -2
params.yml
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
data: cnn_dailymail
|
2 |
-
batch_size:
|
3 |
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
|
|
|
1 |
data: cnn_dailymail
|
2 |
+
batch_size: 2
|
3 |
num_workers: 2
|
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
|