Warning for inception_v3

definitelynotmcarilli · definitelynotmcarilli · commit 7b245dba17f4 · 2019-04-29T10:47:36.000-07:00
diff --git a/examples/imagenet/main_amp.py b/examples/imagenet/main_amp.py
@@ -190,8 +190,9 @@ def resume():
     valdir = os.path.join(args.data, 'val')
 
     if(args.arch == "inception_v3"):
-        crop_size = 299
-        val_size = 320 # I chose this value arbitrarily, we can adjust.
+        raise RuntimeError("Currently, inception_v3 is not supported by this example.")
+        # crop_size = 299
+        # val_size = 320 # I chose this value arbitrarily, we can adjust.
     else:
         crop_size = 224
         val_size = 256
