| { | |
| "batch_size": 2, | |
| "gradient_accumulation_steps": 4, | |
| "epochs": 200, | |
| "lr": 5e-05, | |
| "backbone_lr": 1e-05, | |
| "weight_decay": 0.0001, | |
| "backbone_type": "resnet50", | |
| "use_pretrained": true, | |
| "freeze_backbone_epochs": 5, | |
| "use_gradient_checkpointing": true, | |
| "image_size": [ | |
| 1920, | |
| 1080 | |
| ], | |
| "num_classes": 11, | |
| "merge_banknotes": true, | |
| "train_ratio": 0.85, | |
| "test_ratio": 0.15, | |
| "fpn_channels": 256, | |
| "strides": [ | |
| 8, | |
| 16, | |
| 32, | |
| 64, | |
| 128 | |
| ], | |
| "use_native_resolution": true, | |
| "focal_alpha": 0.5, | |
| "focal_gamma": 2.5, | |
| "lambda_reg": 2.0, | |
| "lambda_ctr": 1.0, | |
| "nms_threshold": 0.4, | |
| "score_threshold": 0.35, | |
| "max_detections": 100, | |
| "patience": 30, | |
| "grad_clip": 1.0, | |
| "warmup_epochs": 10, | |
| "hf_dataset": "format37/russian-rubles-banknotes", | |
| "hf_model_repo": "format37/fcos-banknotes-detector" | |
| } |