fixed network file initialization for GPU device

quasi-robotics · Dec 3, 2021 · 9ccc8d5 · 9ccc8d5
1 parent 26340b6
commit 9ccc8d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gpd/net/openvino_classifier.cpp b/src/gpd/net/openvino_classifier.cpp
@@ -14,7 +14,7 @@ namespace
       out_model_xml = std::string(MODELS_DIR) + "/fp32/single_view_15_channels.xml";
       out_trained_file = std::string(MODELS_DIR) + "/fp32/single_view_15_channels.bin";
     }
-    else if(device == Classifier::Device::eVPU || device == Classifier::Device::eVPU)
+    else if(device == Classifier::Device::eGPU || device == Classifier::Device::eVPU)
     {
       out_model_xml = std::string(MODELS_DIR) + "/fp16/single_view_15_channels.xml";
       out_trained_file = std::string(MODELS_DIR) + "/fp16/single_view_15_channels.bin";