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

restore --version flag in resnet parser

restore --version flag in resnet parser (#3597)
restore --version flag in resnet parser
5d386d5b · Taylor Robie · GitHub · 086d9148 · 5d386d5b
Unverified Commit 5d386d5b authored Mar 14, 2018 by Taylor Robie Committed by GitHub Mar 14, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

official/resnet/resnet.py official/resnet/resnet.py +6 -0

No files found.
--- a/official/resnet/resnet.py
+++ b/official/resnet/resnet.py
@@ -787,6 +787,12 @@ class ResnetArgParser(argparse.ArgumentParser):
        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(
        '--resnet_size', '-rs', type=int, default=50,
        choices=resnet_size_choices,