Skip to content

GitLab

Unverified Commit 5d386d5b authored by Taylor Robie's avatar Taylor Robie Committed by GitHub
Browse files

restore --version flag in resnet parser (#3597) 

restore --version flag in resnet parser
parent 086d9148
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
official/resnet/resnet.py
View file @ 5d386d5b
...@@ -787,6 +787,12 @@ class ResnetArgParser(argparse.ArgumentParser): ...@@ -787,6 +787,12 @@ class ResnetArgParser(argparse.ArgumentParser):
parsers.ImageModelParser(), parsers.ImageModelParser(),
]) ])
self.add_argument(
'--version', '-v', type=int, choices=[1, 2],
default=DEFAULT_VERSION,
help="Version of ResNet. (1 or 2) See README.md for details."
)
self.add_argument( self.add_argument(
'--resnet_size', '-rs', type=int, default=50, '--resnet_size', '-rs', type=int, default=50,
choices=resnet_size_choices, choices=resnet_size_choices,
......
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