t5-small-squadshifts-new_wiki-qg / trainer_config.json
asahi417's picture
update
a7bb9e4
raw
history blame
365 Bytes
{"dataset_path": "lmqg/qg_squadshifts", "dataset_name": "new_wiki", "input_types": ["paragraph_answer"], "output_types": ["question"], "prefix_types": ["qg"], "model": "lmqg/t5-small-squad", "max_length": 512, "max_length_output": 32, "epoch": 6, "batch": 32, "lr": 0.0001, "fp16": false, "random_seed": 1, "gradient_accumulation_steps": 2, "label_smoothing": 0.15}