Warning for inception_v3

7b245dba · Michael Carilli · 855808f3 · 7b245dba
Commit 7b245dba authored Apr 29, 2019 by Michael Carilli
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

examples/imagenet/main_amp.py examples/imagenet/main_amp.py +3 -2

No files found.
--- a/examples/imagenet/main_amp.py
+++ b/examples/imagenet/main_amp.py
@@ -190,8 +190,9 @@ def main():
    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