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