config.json

{
  "accelerator": "auto",
  "architecture": "retinanet",
  "batch_size": 1,
  "devices": "auto",
  "nms_thresh": 0.05,
  "num_classes": 1,
  "retinanet": {
    "score_thresh": 0.1
  },
  "train": {
    "csv_file": null,
    "epochs": 1,
    "fast_dev_run": false,
    "lr": 0.001,
    "preload_images": false,
    "root_dir": null
  },
  "validation": {
    "csv_file": null,
    "iou_threshold": 0.4,
    "root_dir": null,
    "val_accuracy_interval": 20
  },
  "workers": 1
}