trainer_config.json

{
  "dataset_path": "asahi417/qg_squad",
  "dataset_name": "default",
  "input_types": [
    "sentence_answer",
  ],
  "output_types": [
    "question"
  ],
  "prefix_types": null,
  "model": "facebook/bart-base",
  "max_length": 128,
  "max_length_output": 32,
  "epoch": 3,
  "batch": 64,
  "lr": 0.0001,
  "fp16": false,
  "random_seed": 1,
  "gradient_accumulation_steps": 2,
  "label_smoothing": 0.15
}