yolov5下载之后运行train.py出现问题

Question

提示代码为：
Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have 1024 channels, but got 512 channels instead
请问该如何解决