| { |
| "train_config": { |
| "lr": 0.0001, |
| "lr_encoder": 0.00015, |
| "batch_size": 9, |
| "grad_accum_steps": 2, |
| "auto_batch_target_effective": 16, |
| "auto_batch_max_targets_per_image": 100, |
| "auto_batch_ema_headroom": 0.7, |
| "epochs": 500, |
| "resume": null, |
| "ema_decay": 0.993, |
| "ema_tau": 100, |
| "lr_drop": 100, |
| "checkpoint_interval": 100, |
| "skip_best_epochs": 0, |
| "smooth_alpha": 0.0, |
| "warmup_epochs": 0.0, |
| "lr_vit_layer_decay": 0.8, |
| "lr_component_decay": 0.7, |
| "drop_path": 0.0, |
| "cls_loss_coef": 1.0, |
| "keypoint_flip_pairs": [], |
| "keypoint_l1_loss_coef": 0, |
| "keypoint_findable_loss_coef": 0, |
| "keypoint_visible_loss_coef": 0, |
| "keypoint_nll_loss_coef": 0, |
| "keypoint_oks_sigmas": null, |
| "dataset_file": "roboflow", |
| "square_resize_div_64": true, |
| "dataset_dir": "/home/saumya.saksena/projects/ExDark/data", |
| "output_dir": "/home/saumya.saksena/projects/DetectionBench/experiments/exdark/rfdetr-medium", |
| "multi_scale": true, |
| "expanded_scales": true, |
| "do_random_resize_via_padding": false, |
| "use_ema": true, |
| "ema_update_interval": 1, |
| "eval_ema_only": false, |
| "num_workers": 4, |
| "weight_decay": 0.0001, |
| "amp_dtype": "auto", |
| "early_stopping": true, |
| "early_stopping_patience": 100, |
| "early_stopping_min_delta": 0.001, |
| "early_stopping_use_ema": false, |
| "progress_bar": "rich", |
| "tensorboard": true, |
| "wandb": false, |
| "mlflow": false, |
| "clearml": false, |
| "project": null, |
| "run": null, |
| "class_names": null, |
| "run_test": false, |
| "eval_max_dets": 500, |
| "eval_interval": 1, |
| "log_per_class_metrics": true, |
| "eval_masks_head_resolution": false, |
| "aug_config": null, |
| "scale_jitter": true, |
| "augmentation_backend": "kornia", |
| "save_dataset_grids": false, |
| "notes": null, |
| "accelerator": "auto", |
| "clip_max_norm": 0.1, |
| "seed": 42, |
| "sync_bn": false, |
| "strategy": "auto", |
| "devices": 1, |
| "num_nodes": 1, |
| "fp16_eval": false, |
| "lr_scheduler": "step", |
| "lr_scheduler_kwargs": {}, |
| "lr_scheduler_interval": "step", |
| "lr_scheduler_monitor": "val/loss", |
| "lr_min_factor": 0.0, |
| "optimizer": "adamw", |
| "optimizer_kwargs": {}, |
| "dont_save_weights": false, |
| "train_log_sync_dist": false, |
| "train_log_on_step": false, |
| "compute_train_metrics": false, |
| "compute_val_loss": true, |
| "compute_test_loss": true, |
| "pin_memory": null, |
| "persistent_workers": null, |
| "prefetch_factor": null |
| }, |
| "model_config": { |
| "encoder": "dinov2_windowed_small", |
| "out_feature_indexes": [ |
| 3, |
| 6, |
| 9, |
| 12 |
| ], |
| "dec_layers": 4, |
| "two_stage": true, |
| "projector_scale": [ |
| "P4" |
| ], |
| "hidden_dim": 256, |
| "patch_size": 16, |
| "num_windows": 2, |
| "sa_nheads": 8, |
| "ca_nheads": 16, |
| "dec_n_points": 2, |
| "num_queries": 300, |
| "num_select": 300, |
| "postprocess_trace_alpha": 0.2, |
| "bbox_reparam": true, |
| "lite_refpoint_refine": true, |
| "layer_norm": true, |
| "amp": true, |
| "num_channels": 3, |
| "num_classes": 12, |
| "pretrain_weights": "/home/saumya.saksena/.roboflow/models/rf-detr-medium.pth", |
| "device": "cuda", |
| "resolution": 576, |
| "group_detr": 13, |
| "gradient_checkpointing": false, |
| "compile": false, |
| "fused_optimizer": true, |
| "positional_encoding_size": 36, |
| "ia_bce_loss": true, |
| "segmentation_head": false, |
| "use_grouppose_keypoints": false, |
| "keypoint_cross_attn": true, |
| "inter_instance_kp_attn": false, |
| "grouppose_keypoint_dim_downscale": 1, |
| "dual_projector": false, |
| "dual_projector_kp_only": false, |
| "num_keypoints_per_class": [], |
| "num_decoder_registers": 0, |
| "mask_downsample_ratio": 4, |
| "backbone_lora": false, |
| "freeze_encoder": false, |
| "license": "Apache-2.0", |
| "model_name": "RFDETRMedium" |
| }, |
| "model_config_type": "RFDETRMediumConfig", |
| "class_names": [ |
| "Bicycle", |
| "Boat", |
| "Bottle", |
| "Bus", |
| "Car", |
| "Cat", |
| "Chair", |
| "Cup", |
| "Dog", |
| "Motorbike", |
| "People", |
| "Table" |
| ], |
| "num_classes": 12 |
| } |
