gagan3012 commited on
Commit
7e7efe5
·
1 Parent(s): 940c82c
Files changed (1) hide show
  1. params.yml +2 -2
params.yml CHANGED
@@ -1,10 +1,10 @@
1
  data: cnn_dailymail
2
- batch_size: 4
3
  num_workers: 2
4
  model_type: t5
5
  model_name: t5-small
6
  learning_rate: 1e-4
7
- epochs: 5
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