| { | |
| "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 | |
| } |