forked from mindspore-Ecosystem/mindspore
ssd300 performance fix
This commit is contained in:
parent
00ac7b855c
commit
c56605bcf8
|
@ -67,7 +67,6 @@ 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->SetPrecisionMode("allow_fp32_to_fp16");
|
||||
context->MutableDeviceInfo().push_back(ascend310);
|
||||
mindspore::Graph graph;
|
||||
Serialization::Load(FLAGS_mindir_path, ModelType::kMindIR, &graph);
|
||||
|
|
Loading…
Reference in New Issue