summarization / model_params.yml
Dean
PR conflict resolution plus making HF upload more generic
09be2fb
raw history blame
No virus
177 Bytes
name: summarsiation
model_type: t5
model_name: t5-small
batch_size: 2
epochs: 5
use_gpu: True
learning_rate: 1e-4
num_workers: 2
model_dir: models
metric: rouge
source_dir: src