Repository link in readme
Browse files
README.md
CHANGED
@@ -72,3 +72,6 @@ The model was trained for 140 epochs on 1 RTX 4060 Ti GPU(Finetuning decoder onl
|
|
72 |
|
73 |
This model achieves an mAP (average precision) of **59.0** on SKU110k validation set. Result was calculated with torchmetrics MeanAveragePrecision class.
|
74 |
|
|
|
|
|
|
|
|
72 |
|
73 |
This model achieves an mAP (average precision) of **59.0** on SKU110k validation set. Result was calculated with torchmetrics MeanAveragePrecision class.
|
74 |
|
75 |
+
## Training Code
|
76 |
+
|
77 |
+
Code is released in this repository [Repo Link](https://github.com/Isalia20/DETR-finetune/tree/main). However it's not finalized/tested well yet but the main stuff is in the code.
|