shape '[1024, 512, 3, 3]' is invalid for input of size 4178590

[miscla]

Traceback (most recent call last): File "detect.py", line 118, in model.load_weights(args.weightsfile) File "D:\Skripsi\pytorch-yolo-v3\darknet.py", line 477, in load_weights conv_weights = conv_weights.view_as(conv.weight.data) RuntimeError: shape '[1024, 512, 3, 3]' is invalid for input of size 4178590

i have an error like that.. i only have 1 class, so i changed classes in .cfg to 1 and filter in .cfg to 18 iam using my own .weight from training using darknet

anyone have solution for my problem??