File size: 205 Bytes
4804221
71416a8
b0479bb
e76c5c8
deaf5ba
52fb674
26222c4
1a42b6a
23fb59e
4f3c5d9
6d67366
69ccc27
c7b9826
1
2
3
4
5
6
7
8
9
10
11
12
13
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
metric: rouge
split: 0.001
use_gpu: True
visualise: True