Explorar el Código

preprocessing fn argument check

John Welsh hace 7 años
padre
commit
54aac518c8
Se han modificado 1 ficheros con 4 adiciones y 1 borrados
  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;