From c9028157ed4bd8860d3a9dd32e926966312b8ea6 Mon Sep 17 00:00:00 2001
From: yuzhenhua <yuzhenhua.yuzhenhua@huawei.com>
Date: Wed, 31 Mar 2021 14:55:35 +0800
Subject: [PATCH] fix ssd 310 infer acc bug

---
 model_zoo/official/cv/ssd/ascend310_infer/src/main.cc | 1 +
 1 file changed, 1 insertion(+)

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<Context>();
   auto ascend310 = std::make_shared<mindspore::Ascend310DeviceInfo>();
   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);