Traceback (most recent call last):


 "train.py", line 64, in <module>

model.load_darknet_weights(opt.pretrained_weights)


 "/home/data/PyTorch-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 '[512, 256, 3, 3]' is invalid for input of size 391694

Traceback (most recent call last):


 "/home/zuanazzi/PointCloudFlow/pn2_test.py", line 25, in <module>

opt.step()


 "/home/zuanazzi/anaconda3/envs/kaolin/lib/python3.6/site-packages/torch/optim/adam.py", line 96, in step

exp_avg_sq.mul_(beta2).addcmul_(1 - beta2, grad, grad)
RuntimeError: CUDA error: an illegal memory access was encountered
