Hugging Face's logo

Spaces:
gagan3012
/
summarization
Runtime error

App Files Community
gagan3012 commited on
Commit
84ffbe6
·
1 Parent(s): f9d6102

fixes

Browse files
Files changed (1) hide show
  1. params.yml +2 -2
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: 3
8
  source_dir: src
9
  model_dir: models
10
  metric: rouge
11
- split: 0.01
12
  use_gpu: True
 
4
  model_type: t5
5
  model_name: t5-small
6
  learning_rate: 1e-4
7
+ epochs: 10
8
  source_dir: src
9
  model_dir: models
10
  metric: rouge
11
+ split: 0.05
12
  use_gpu: True