gooseable

一次训练到11000次，在train.py中valid_in_image[index,:,:,:]处报错，错误为 IndexError: index 7 is out of bounds for axis 0 with size 7 。改变验证集大小后，训练到12000次中断，报同样的错误。请问这个怎么解决？

训练时model.py在load_darknet_weights处报错

3

您好，我进行训练的时候报以下错误 File "train.py", line 77, in model.load_darknet_weights(opt.pretrained_weights) File "/home/cct/YOLOv3-model-pruning/models.py", line 317, in load_darknet_weights conv_w = torch.from_numpy(weights[ptr: ptr + num_w]).view_as(conv_layer.weight) RuntimeError: shape '[256, 128, 3, 3]' is invalid for input of...