search

YuHengsss / YOLOV

This repo is an implementation of PyTorch version YOLOV Series
Apache License 2.0
309 stars 43 forks source link

Train Error "IndexError: The shape of the mask [60] at index 0 does not match the shape of the indexed tensor [30, 30] at index 0" #7

Closed LHB116 closed 2 years ago

LHB116 commented 2 years ago

Dear author, thank you very much for sharing your excellent research. It is very innovative and gets outstanding results. I'm trying to training yolov_s model with exp file yolov_s_online.py. But I encountered a problem "IndexError: The shape of the mask [60] at index 0 does not match the shape of the indexed tensor [30, 30] at index 0". Can you tell me how to fix it. Looking forward to your reply. Thank you very much.

YuHengsss commented 2 years ago

Dear author, thank you very much for sharing your excellent research. It is very innovative and gets outstanding results. I'm trying to training yolov_s model with exp file yolov_s_online.py. But I encountered a problem "IndexError: The shape of the mask [60] at index 0 does not match the shape of the indexed tensor [30, 30] at index 0". Can you tell me how to fix it. Looking forward to your reply. Thank you very much.

Please do not use online files for training, they are only used for testing.