about train

[voicccc]

Search before asking

• [X] I have searched the ModelForge issues and found no similar bug report.

YOLOv5 Component

No response

Bug

Hello, run the command: python train.py --cfg models/yolox.yaml --noautoanchor, When the code executes the val verification process, this error occurs,

File "/root/ModelForge-master/models/yolox.py", line 523, in get_assignments
    fg_mask, is_in_boxes_and_center = self.get_in_boxes_info(
  File "/root/ModelForge-master/models/yolox.py", line 636, in get_in_boxes_info
    b_l = x_centers_per_image - gt_bboxes_per_image_l
RuntimeError: The size of tensor a (2000) must match the size of tensor b (1400) at non-singleton dimension 1

How to deal with it?

Environment

No response

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

• [ ] Yes I'd like to help by submitting a PR!