isalia99 commited on
Commit
06a64e5
1 Parent(s): 2d329b8

Repository link in readme

Browse files
Files changed (1) hide show
  1. README.md +3 -0
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.