Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #555 from MichaelMonashev/patch-1

Browse Source 
benchmark.py argument description fixed
pull/612/head
Ross Wightman 4 years ago committed by GitHub
parent 67d0665b46 0be1fa4793
commit 137a374930
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
benchmark.py
Unescape View File

@ -48,7 +48,7 @@ parser = argparse.ArgumentParser(description='PyTorch Benchmark')
parser.add_argument('--model-list', metavar='NAME', default='', parser.add_argument('--model-list', metavar='NAME', default='',
help='txt file based list of model names to benchmark') help='txt file based list of model names to benchmark')
parser.add_argument('--bench', default='both', type=str, parser.add_argument('--bench', default='both', type=str,
help="Benchmark mode. One of 'inference', 'train', 'both'. Defaults to 'inference'") help="Benchmark mode. One of 'inference', 'train', 'both'. Defaults to 'both'")
parser.add_argument('--detail', action='store_true', default=False, parser.add_argument('--detail', action='store_true', default=False,
help='Provide train fwd/bwd/opt breakdown detail if True. Defaults to False') help='Provide train fwd/bwd/opt breakdown detail if True. Defaults to False')
parser.add_argument('--results-file', default='', type=str, metavar='FILENAME', parser.add_argument('--results-file', default='', type=str, metavar='FILENAME',

Write Preview
Loading…
Cancel
Save