lucasjinreal / yolov7_d2

🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥
GNU General Public License v3.0
3.13k stars 483 forks source link

key error yolomask.py #188

Open JISHNUSHAJI opened 1 year ago

JISHNUSHAJI commented 1 year ago

I got the following error while trying to initiate training.Any solution?

File "/home/yolov7/yolov7_d2/yolov7/modeling/meta_arch/yolomask.py", line 270, in forward for key in losses[0].keys(): AttributeError: 'tuple' object has no attribute 'keys'