Hugging Face's logo

phucn001
/
hangcon-rfdetr-medium

Object Detection
TensorBoard
rfdetr
computer-vision
Model card Files
xet
 Metrics Community
phucn001 commited on
Commit
2db062e
·
verified ·
1 Parent(s): bdb1a72

Upload artifact_manifest.json for hangcon-rfdetr-h100-medium-672-bs32-ga1

Browse files
Files changed (1) hide show
  1. artifact_manifest.json +178 -0
artifact_manifest.json ADDED
@@ -0,0 +1,178 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "created_at_utc": "2026-08-03T14:07:20.771124+00:00",
3
+ "backend": "rfdetr",
4
+ "experiment_name": "hangcon-rfdetr-h100-medium-672-bs32-ga1",
5
+ "model_path": "rfdetr-medium",
6
+ "selected_model": "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/checkpoint_best_total.pth",
7
+ "selected_model_name": "checkpoint_best_total.pth",
8
+ "selected_model_size_mb": 128.19,
9
+ "resume_only_ckpt_files": [
10
+ "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/checkpoint_24.ckpt",
11
+ "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/checkpoint_49.ckpt",
12
+ "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/last.ckpt"
13
+ ],
14
+ "export_formats": [
15
+ "onnx"
16
+ ],
17
+ "exported_files": [],
18
+ "export_errors": [],
19
+ "event_files": [
20
+ "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/events.out.tfevents.1785683262.vision-lab-rf-detr-z1wohc5m-54c5bd99df-cnd4x.5717.0",
21
+ "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/events.out.tfevents.1785683592.vision-lab-rf-detr-z1wohc5m-54c5bd99df-cnd4x.6111.0",
22
+ "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/events.out.tfevents.1785683607.vision-lab-rf-detr-z1wohc5m-54c5bd99df-cnd4x.6305.0",
23
+ "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1/events.out.tfevents.1785683999.vision-lab-rf-detr-z1wohc5m-54c5bd99df-cnd4x.6810.0"
24
+ ],
25
+ "run_dir": "/workspace/runs/hangcon-rfdetr-h100-medium-672-bs32-ga1",
26
+ "checkpoint_run_dir": "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1",
27
+ "export_run_dir": "/workspace/export/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1",
28
+ "checkpoint_backup_run_dir": "/workspace/checkpoint_backup/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1",
29
+ "dataset_path": "/workspace/HangCon_RFDETR_COCO",
30
+ "dataset_yaml": "/workspace/HangCon/data.yaml",
31
+ "image_size": 672,
32
+ "batch_size": 32,
33
+ "device": "0",
34
+ "training_config": {
35
+ "train_config": {
36
+ "lr": 0.0001,
37
+ "lr_encoder": 0.00015,
38
+ "batch_size": 32,
39
+ "grad_accum_steps": 1,
40
+ "auto_batch_target_effective": 16,
41
+ "auto_batch_max_targets_per_image": 100,
42
+ "auto_batch_ema_headroom": 0.7,
43
+ "epochs": 117,
44
+ "resume": null,
45
+ "ema_decay": 0.993,
46
+ "ema_tau": 100,
47
+ "lr_drop": 100,
48
+ "checkpoint_interval": 25,
49
+ "skip_best_epochs": 0,
50
+ "smooth_alpha": 0.0,
51
+ "warmup_epochs": 0.0,
52
+ "lr_vit_layer_decay": 0.8,
53
+ "lr_component_decay": 0.7,
54
+ "drop_path": 0.0,
55
+ "group_detr": 13,
56
+ "ia_bce_loss": true,
57
+ "cls_loss_coef": 1.0,
58
+ "num_select": 300,
59
+ "keypoint_flip_pairs": [],
60
+ "keypoint_l1_loss_coef": 0,
61
+ "keypoint_findable_loss_coef": 0,
62
+ "keypoint_visible_loss_coef": 0,
63
+ "keypoint_nll_loss_coef": 0,
64
+ "keypoint_oks_sigmas": null,
65
+ "dataset_file": "roboflow",
66
+ "square_resize_div_64": true,
67
+ "dataset_dir": "/workspace/HangCon_RFDETR_COCO",
68
+ "output_dir": "/workspace/model_checkpoints/rfdetr/hangcon-rfdetr-h100-medium-672-bs32-ga1",
69
+ "multi_scale": false,
70
+ "expanded_scales": true,
71
+ "do_random_resize_via_padding": false,
72
+ "use_ema": true,
73
+ "ema_update_interval": 1,
74
+ "num_workers": 12,
75
+ "weight_decay": 0.0001,
76
+ "amp_dtype": "auto",
77
+ "early_stopping": true,
78
+ "early_stopping_patience": 50,
79
+ "early_stopping_min_delta": 0.001,
80
+ "early_stopping_use_ema": false,
81
+ "progress_bar": null,
82
+ "tensorboard": true,
83
+ "wandb": false,
84
+ "mlflow": false,
85
+ "clearml": false,
86
+ "project": null,
87
+ "run": null,
88
+ "class_names": null,
89
+ "run_test": false,
90
+ "segmentation_head": false,
91
+ "eval_max_dets": 500,
92
+ "eval_interval": 1,
93
+ "log_per_class_metrics": true,
94
+ "aug_config": null,
95
+ "augmentation_backend": "cpu",
96
+ "save_dataset_grids": false,
97
+ "notes": null,
98
+ "accelerator": "auto",
99
+ "clip_max_norm": 0.1,
100
+ "seed": null,
101
+ "sync_bn": false,
102
+ "strategy": "auto",
103
+ "devices": 1,
104
+ "num_nodes": 1,
105
+ "fp16_eval": false,
106
+ "lr_scheduler": "step",
107
+ "lr_min_factor": 0.0,
108
+ "dont_save_weights": false,
109
+ "train_log_sync_dist": false,
110
+ "train_log_on_step": false,
111
+ "compute_train_metrics": false,
112
+ "compute_val_loss": true,
113
+ "compute_test_loss": true,
114
+ "pin_memory": null,
115
+ "persistent_workers": null,
116
+ "prefetch_factor": null
117
+ },
118
+ "model_config": {
119
+ "encoder": "dinov2_windowed_small",
120
+ "out_feature_indexes": [
121
+ 3,
122
+ 6,
123
+ 9,
124
+ 12
125
+ ],
126
+ "dec_layers": 4,
127
+ "two_stage": true,
128
+ "projector_scale": [
129
+ "P4"
130
+ ],
131
+ "hidden_dim": 256,
132
+ "patch_size": 16,
133
+ "num_windows": 2,
134
+ "sa_nheads": 8,
135
+ "ca_nheads": 16,
136
+ "dec_n_points": 2,
137
+ "num_queries": 300,
138
+ "num_select": 300,
139
+ "postprocess_trace_alpha": 0.2,
140
+ "bbox_reparam": true,
141
+ "lite_refpoint_refine": true,
142
+ "layer_norm": true,
143
+ "amp": true,
144
+ "num_channels": 3,
145
+ "num_classes": 2,
146
+ "pretrain_weights": "/root/.roboflow/models/rf-detr-medium.pth",
147
+ "device": "cuda",
148
+ "resolution": 672,
149
+ "group_detr": 13,
150
+ "gradient_checkpointing": false,
151
+ "compile": false,
152
+ "fused_optimizer": true,
153
+ "positional_encoding_size": 42,
154
+ "ia_bce_loss": true,
155
+ "cls_loss_coef": 1.0,
156
+ "segmentation_head": false,
157
+ "use_grouppose_keypoints": false,
158
+ "keypoint_cross_attn": true,
159
+ "inter_instance_kp_attn": false,
160
+ "grouppose_keypoint_dim_downscale": 1,
161
+ "dual_projector": false,
162
+ "dual_projector_kp_only": false,
163
+ "num_keypoints_per_class": [],
164
+ "num_decoder_registers": 0,
165
+ "mask_downsample_ratio": 4,
166
+ "backbone_lora": false,
167
+ "freeze_encoder": false,
168
+ "license": "Apache-2.0",
169
+ "model_name": "RFDETRMedium"
170
+ },
171
+ "model_config_type": "RFDETRMediumConfig",
172
+ "class_names": [
173
+ "hanging_object",
174
+ "hanging_rope"
175
+ ],
176
+ "num_classes": 2
177
+ }
178
+ }