提交 3e839629 编写于 作者: H HappyAngel 提交者: yiicy

fiix conv_bn fuse error in model_test, test=develop (#2531)

上级 4fc0dfe2
......@@ -44,10 +44,7 @@ void OutputOptModel(const std::string& load_model_dir,
const std::vector<std::vector<int64_t>>& input_shapes) {
lite_api::CxxConfig config;
config.set_model_dir(load_model_dir);
std::vector<Place> vaild_places = {
Place{TARGET(kARM), PRECISION(kFloat)},
Place{TARGET(kX86), PRECISION(kFloat)},
};
std::vector<Place> vaild_places = {Place{TARGET(kARM), PRECISION(kFloat)}};
if (FLAGS_is_quantized_model) {
vaild_places.insert(vaild_places.begin(),
Place{TARGET(kARM), PRECISION(kInt8)});
......
......@@ -46,10 +46,7 @@ void OutputOptModel(const std::string& load_model_dir,
const std::vector<std::vector<int64_t>>& input_shapes) {
lite_api::CxxConfig config;
config.set_model_dir(load_model_dir);
config.set_valid_places({
Place{TARGET(kX86), PRECISION(kFloat)},
Place{TARGET(kARM), PRECISION(kFloat)},
});
config.set_valid_places({Place{TARGET(kARM), PRECISION(kFloat)}});
auto predictor = lite_api::CreatePaddlePredictor(config);
// delete old optimized model
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册