File size: 196 Bytes
4804221
71416a8
b0479bb
e76c5c8
deaf5ba
52fb674
26222c4
1a42b6a
52efc84
4f3c5d9
8eb4aff
102f69c
1
2
3
4
5
6
7
8
9
10
11
12
data: cnn_dailymail
batch_size: 2
num_workers: 2
model_type: t5
model_name: t5-small
learning_rate: 1e-4
epochs: 5
source_dir: src
model_dir: models/models/
metric: rouge
split: 0.02
use_gpu: True