Преглед на файлове

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;