When I run train.py in PYCHARM, errors occured as follows:

[longlizhi]

/home/hndx/anaconda3/envs/yolo6d-1/bin/python /home/hndx/11z/singleshotpose-master-1/train. py
Traceback (most recent calt Last):
File"/hame/hngx/LLz/singleshotpose-master-/trainpy", Line 338, in
modeL. Load_weights_until_last(initweightfile)
FiLe"/home/hngx/u2/sinqleshotpose-master-/darknetpy", line 321, in Load_weights_until_Last start=load_conv_bn(buf, start, model[
el, model[1])
File "/home/hngx/1z/sinqieshotpose-master-1/ofq. py", tine 175, in 1oad_. conv_hn cony_modeL. weight. data. copy_(torch.
from_numpy (buf[ start: start+num-wl). view_as(conv_modeL. weight. data)); start =start+num-w
RuntimeError: shape'[128,64,3,3]' is invalid for input of size 69839