RuntimeError: shape '[256, 128, 3, 3]' is invalid for input of size 81624

[dygcjlu]

I follow the guide to do the configuration, and when i start to train the model (python examples/train.py Yolov3), the following error occured.
What can i do to fixed this issue?

Error:
File "./vedanet/models/_darknet.py", line 155, in _load_convbatch
.view_as(model.layers[0].weight.data))
RuntimeError: shape '[256, 128, 3, 3]' is invalid for input of size 81624