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

remove gpu memory alloc

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