dog-qiuqiu / FastestDet

:zap: A newly designed ultra lightweight anchor free target detection algorithm, weight only 250K parameters, reduces the time consumption by 10% compared with yolo-fastest, and the post-processing is simpler
BSD 3-Clause "New" or "Revised" License
768 stars 131 forks source link

MemoryError #27

Closed centurions closed 1 year ago

centurions commented 1 year ago

Hi I get this error with batchsize = 32 after 10 epoch creating index... index created! Traceback (most recent call last): File "train.py", line 134, in model.train() File "train.py", line 126, in train mAP05 = self.evaluation.compute_map(self.val_dataloader, self.model) File "C:\final\dataset\FastestDet-main\utils\evaluation.py", line 98, in compute_map mAP05 = self.coco_evaluate(gts, pts) File "C:\final\dataset\FastestDet-main\utils\evaluation.py", line 45, in coco_evaluate coco_pred.dataset["images"].append({"id": i}) MemoryError