RuntimeError: shape '[1024, 512, 3, 3]

[swerizwan]

Traceback (most recent call last):
File "demo.py", line 573, in
run_video_demo(args)
File "demo.py", line 224, in run_video_demo
mot = MPT(
File "/home/abbas/anaconda3/envs/pymafx/lib/python3.8/site-packages/multi_person_tracker/mpt.py", line 52, in init
self.detector = YOLOv3(
File "/home/abbas/anaconda3/envs/pymafx/lib/python3.8/site-packages/yolov3/yolo.py", line 51, in init
self.model.load_darknet_weights(weights_path)
File "/home/abbas/anaconda3/envs/pymafx/lib/python3.8/site-packages/yolov3/models.py", line 315, in load_darknet_weights
conv_w = torch.from_numpy(weights[ptr : ptr + num_w]).view_as(conv_layer.weight)
RuntimeError: shape '[1024, 512, 3, 3]' is invalid for input of size 2112558
(pymafx) abbas@amax-22:~/PYMAF/PyMAF-smpl$ ^C