形状“[1024, 1024, 1, 1]”对于大小为 510554 的输入无效

Question

--> conv_model.weight.data.copy_(torch.from_numpy(buf[start:start + num_w]).reshape(conv_model.weight.data.shape)); 运行时错误：形状“[1024, 1024, 1, 1]”对于大小为 510554 的输入无效

我正在尝试将 .weights 转换为 .pt 或 .onnx 格式。