提交 7dfe3bdf 编写于 作者: G gaoyuan

remove gpu memory alloc

上级 38723e77
......@@ -53,7 +53,7 @@ void doOnePriorBoxTest(size_t featureMapWidth,
LayerMap layerMap;
vector<Argument> datas;
initDataLayer(
configt, &dataLayers, &datas, &layerMap, "priorbox", 1, false, true);
configt, &dataLayers, &datas, &layerMap, "priorbox", 1, false, false);
dataLayers[0]->getOutput().setFrameHeight(featureMapHeight);
dataLayers[0]->getOutput().setFrameWidth(featureMapWidth);
dataLayers[1]->getOutput().setFrameHeight(imageHeight);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册