mbart-large-cc25-esquad-qg-ae / trainer_config.json
asahi417's picture
model update
1be914d
raw
history blame
403 Bytes
{"dataset_path": "lmqg/qg_esquad", "dataset_name": "default", "input_types": ["paragraph_answer", "paragraph_sentence"], "output_types": ["question", "answer"], "prefix_types": ["qg", "ae"], "model": "facebook/mbart-large-cc25", "max_length": 512, "max_length_output": 32, "epoch": 5, "batch": 2, "lr": 0.0001, "fp16": false, "random_seed": 1, "gradient_accumulation_steps": 32, "label_smoothing": 0.15}