Parcourir la source

preprocessing fn argument check

John Welsh il y a 7 ans
Parent
commit
54aac518c8
1 fichiers modifiés avec 4 ajouts et 1 suppressions
  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;