search

deepinsight / insightface

State-of-the-art 2D and 3D Face Analysis Project
https://insightface.ai
23.31k stars 5.41k forks source link

scrfd train error #1541

Closed staystay closed 3 years ago

staystay commented 3 years ago

训练scrfd时，config用scrfd_10g_bnkps.py，报错如下： mmdet/models/losses/iou_loss.py", line 365, in forward return (pred * weight).sum() # 0 RuntimeError: The size of tensor a (4) must match the size of tensor b (8) at non-singleton dimension 1 请问这怎么解决呀？感谢 @nttstar

xun-0612 commented 3 years ago

https://github.com/open-mmlab/mmdetection/pull/4982 看看這個

staystay commented 3 years ago

open-mmlab/mmdetection#4982 看看這個

谢谢，解决啦