Gagan Bhatia commited on
Commit
deaf5ba
1 Parent(s): 95f692c

Update params.yml

Browse files
Files changed (1) hide show
  1. params.yml +1 -0
params.yml CHANGED
@@ -2,6 +2,7 @@ data: cnn_dailymail
2
  batch_size: 2
3
  num_workers: 2
4
  model_type: t5
 
5
  learning_rate: 1e-4
6
  epochs: 5
7
  source_dir: src
 
2
  batch_size: 2
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