From 0be1fa4793aed6d926dff43e59491588c744bd89 Mon Sep 17 00:00:00 2001 From: Michael Monashev Date: Sun, 11 Apr 2021 18:08:43 +0300 Subject: [PATCH] Argument description fixed --- benchmark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmark.py b/benchmark.py index 4812d85c..dc76550a 100755 --- a/benchmark.py +++ b/benchmark.py @@ -48,7 +48,7 @@ parser = argparse.ArgumentParser(description='PyTorch Benchmark') parser.add_argument('--model-list', metavar='NAME', default='', help='txt file based list of model names to benchmark') 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, help='Provide train fwd/bwd/opt breakdown detail if True. Defaults to False') parser.add_argument('--results-file', default='', type=str, metavar='FILENAME',