T5-Summarization / params.yml
Gagan Bhatia
Update params.yml
914d7ff
raw history blame
No virus
267 Bytes
name: summarsiation
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
hf_username: gagan3012
upload_to_hf: True