Skip to content

S
seminar-breakout
Commit 4e9e35b5 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix lint

parent 6d921e36
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/ResNet/imagenet-resnet.py
View file @ 4e9e35b5
...@@ -116,8 +116,8 @@ if __name__ == '__main__': ...@@ -116,8 +116,8 @@ if __name__ == '__main__':
parser.add_argument('-d', '--depth', help='resnet depth', parser.add_argument('-d', '--depth', help='resnet depth',
type=int, default=18, choices=[18, 34, 50, 101, 152]) type=int, default=18, choices=[18, 34, 50, 101, 152])
parser.add_argument('--eval', action='store_true') parser.add_argument('--eval', action='store_true')
parser.add_argument('--batch', help='total batch size. 32 per GPU gives best accuracy, higher values should be similarly good', parser.add_argument('--batch', default=256, type=int,
default=256, type=int) help='total batch size. 32 per GPU gives best accuracy, higher values should be similarly good')
parser.add_argument('--mode', choices=['resnet', 'preact', 'se'], parser.add_argument('--mode', choices=['resnet', 'preact', 'se'],
help='variants of resnet to use', default='resnet') help='variants of resnet to use', default='resnet')
args = parser.parse_args() args = parser.parse_args()
......
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