瀏覽代碼

preprocessing fn argument check

John Welsh 7 年之前
父節點
當前提交
54aac518c8
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      examples/classify_image/classify_image.cu

+ 4 - 1
examples/classify_image/classify_image.cu

@@ -89,7 +89,10 @@ int main(int argc, char *argv[])
   else if (preprocessingFn == "inception")
     preprocessInception(inputDataHost, inputDims);
   else
-    cout << "Unsupported preprocessing function.  Results may not be correct.\n" << endl;
+  {
+    cout << "Invalid preprocessing function argument, must be vgg or inception. \n" << endl;
+    return 1;
+  }
 
   /* transfer to device */
   float *inputDataDevice, *outputDataDevice;