Spaces:

gagan3012
/

summarization

Runtime error

App Files Files Community

summarization / params.yml

gagan3012's picture

fixes

365fc18 over 3 years ago

188 Bytes

	data: cnn_dailymail
	batch_size: 2
	num_workers: 2
	model_type: t5
	model_name: t5-small
	learning_rate: 1e-4
	epochs: 3
	source_dir: src
	model_dir: models
	metric: rouge
	split: 0.01
	use_gpu: True