Skip to content

GitLab

Commit 7b245dba authored by Michael Carilli's avatar Michael Carilli
Browse files

Warning for inception_v3

parent 855808f3
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
examples/imagenet/main_amp.py
View file @ 7b245dba
...@@ -190,8 +190,9 @@ def main(): ...@@ -190,8 +190,9 @@ def main():
valdir = os.path.join(args.data, 'val') valdir = os.path.join(args.data, 'val')
if(args.arch == "inception_v3"): if(args.arch == "inception_v3"):
crop_size = 299 raise RuntimeError("Currently, inception_v3 is not supported by this example.")
val_size = 320 # I chose this value arbitrarily, we can adjust. # crop_size = 299
# val_size = 320 # I chose this value arbitrarily, we can adjust.
else: else:
crop_size = 224 crop_size = 224
val_size = 256 val_size = 256
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment