This commit is contained in:
Ayush Pandey 2019-02-21 08:36:16 +03:00 committed by Dmitry Kurtaev
parent 8cedc052ca
commit 5c7fe0fe05

View File

@ -584,7 +584,7 @@ void ONNXImporter::populateNet(Net dstNet)
for (int j = 1; j < node_proto.input_size(); j++) {
layerParams.blobs.push_back(getBlob(node_proto, constBlobs, j));
}
layerParams.set("num_output", layerParams.blobs[0].size[1]);
layerParams.set("num_output", layerParams.blobs[0].size[1] * layerParams.get<int>("group", 1));
layerParams.set("bias_term", node_proto.input_size() == 3);
}
else if (layer_type == "Transpose")