Detection head trained on COCO ViT-S (2 epochs, loss cls 0.25 / reg 0.33 / ctr 0.60)
Browse files- det_head.safetensors +3 -0
- det_training_log.json +21 -0
det_head.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0e3fbea046ae7052df5042b9aae0f993e6cef1728f86bf440976ad82c8bbd554
|
| 3 |
+
size 11898884
|
det_training_log.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"head_params": 2910555,
|
| 3 |
+
"feat_dim": 384,
|
| 4 |
+
"resolution": 768,
|
| 5 |
+
"epochs": [
|
| 6 |
+
{
|
| 7 |
+
"epoch": 1,
|
| 8 |
+
"cls_loss": 0.3245142708658809,
|
| 9 |
+
"reg_loss": 0.3868400333578376,
|
| 10 |
+
"ctr_loss": 0.6083796071893346,
|
| 11 |
+
"steps": 14785
|
| 12 |
+
},
|
| 13 |
+
{
|
| 14 |
+
"epoch": 2,
|
| 15 |
+
"cls_loss": 0.25320878397090074,
|
| 16 |
+
"reg_loss": 0.3300947344127764,
|
| 17 |
+
"ctr_loss": 0.6003168236011986,
|
| 18 |
+
"steps": 14785
|
| 19 |
+
}
|
| 20 |
+
]
|
| 21 |
+
}
|