switch CPU

pdkyll · Sep 22, 2021 · 394578f · 394578f
1 parent 5440430
commit 394578f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main_for_pose.cpp b/src/main_for_pose.cpp
@@ -22,7 +22,7 @@ void test_coco_person() {
     const int num_thread = 1;
     const float score_thresh = 0.5;
 
-    TNNDevice device_id = TNNGPU;//运行设备
+    TNNDevice device_id = TNNCPU;//运行设备
     ModelParam model_param = COCO_PERSON_PARAM;//模型参数
     KeyPointDetector *detector = new KeyPointDetector(model_path,
                                                       model_param,