summarization / model_params.yml
gagan3012's picture
Update model_params.yml
ec3086f
batch_size: 2
epochs: 5
learning_rate: 0.0001
metric: rouge
model_dir: models
model_name: t5-base
model_type: t5
name: summarsiation
num_workers: 2
source_dir: src
use_gpu: true