diff --git a/model_zoo/official/cv/ssd/ascend310_infer/src/main.cc b/model_zoo/official/cv/ssd/ascend310_infer/src/main.cc index ca67ddb314..d815cee1a0 100644 --- a/model_zoo/official/cv/ssd/ascend310_infer/src/main.cc +++ b/model_zoo/official/cv/ssd/ascend310_infer/src/main.cc @@ -66,6 +66,7 @@ int main(int argc, char **argv) { auto context = std::make_shared(); auto ascend310 = std::make_shared(); ascend310->SetDeviceID(FLAGS_device_id); + ascend310->SetBufferOptimizeMode("off_optimize"); context->MutableDeviceInfo().push_back(ascend310); mindspore::Graph graph; Serialization::Load(FLAGS_mindir_path, ModelType::kMindIR, &graph);