MegEngine / YOLOX

MegEngine implementation of YOLOX
Apache License 2.0
107 stars 15 forks source link

RuntimeError: assertion `dtype == dst.dtype && dst.is_contiguous()' #4

Open amazingzby opened 3 years ago

amazingzby commented 3 years ago

当输入宽高不一致时报错, 在训练过程中报错,报错时机随缘: yolo_head.py", line 351, in get_assignments bboxes_preds_per_image = bboxes_preds_per_image[fg_mask] RuntimeError: assertion `dtype == dst.dtype && dst.is_contiguous()' failed at ../../../../../../dnn/src/common/elemwise/opr_impl.cpp:281: void megdnn::ElemwiseForward::check_layout_and_broadcast(const TensorLayoutPtrArray&, const megdnn::TensorLayout&)

asdcjt commented 3 years ago

Are you using the pretrain model? I loaded it but it had some mistakes: param head.cls_preds.0.bias shape mismatch, should be (1, 5, 1, 1), get (1, 80, 1, 1) and so on