Commit 80cfc2c8 authored by Joaquín Alori's avatar Joaquín Alori Committed by Francisco Massa
Browse files

Fix comment in default arguments (#1243)

parent 93bceaf2
...@@ -148,7 +148,7 @@ if __name__ == "__main__": ...@@ -148,7 +148,7 @@ if __name__ == "__main__":
parser.add_argument('--epochs', default=13, type=int, metavar='N', parser.add_argument('--epochs', default=13, type=int, metavar='N',
help='number of total epochs to run') help='number of total epochs to run')
parser.add_argument('-j', '--workers', default=4, type=int, metavar='N', parser.add_argument('-j', '--workers', default=4, type=int, metavar='N',
help='number of data loading workers (default: 16)') help='number of data loading workers (default: 4)')
parser.add_argument('--lr', default=0.02, type=float, parser.add_argument('--lr', default=0.02, type=float,
help='initial learning rate, 0.02 is the default value for training ' help='initial learning rate, 0.02 is the default value for training '
'on 8 gpus and 2 images_per_gpu') 'on 8 gpus and 2 images_per_gpu')
......
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