style(main.py): update argparse's help info

Author: zjykzj

mixed_multi_gpu/main.py

@@ -70,14 +70,14 @@ def train(gpu, args):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument('-n', '--nodes', default=1, type=int, metavar='N',
-                        help='number of data loading workers (default: 4)')
     parser.add_argument('-g', '--gpus', default=1, type=int,
-                        help='number of gpus per node')
+                        help='number of gpus per node (default: 1)')
+    parser.add_argument('-n', '--nodes', default=1, type=int, metavar='N',
+                        help='number of machines (default: 1)')
     parser.add_argument('-nr', '--nr', default=0, type=int,
-                        help='ranking within the nodes')
-    parser.add_argument('--epochs', default=2, type=int, metavar='N',
-                        help='number of total epochs to run')
+                        help='ranking within the nodes (default: 0)')
+    parser.add_argument('-e', '--epochs', default=2, type=int, metavar='N',
+                        help='number of total epochs to run (default: 2)')
     args = parser.parse_args()
     # train(0, args)
     args.world_size = args.gpus * args.nodes

mixed_single_gpu/main.py

@@ -68,14 +68,14 @@ def train(gpu, args):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument('-n', '--nodes', default=1, type=int, metavar='N',
-                        help='number of data loading workers (default: 4)')
     parser.add_argument('-g', '--gpus', default=1, type=int,
-                        help='number of gpus per node')
+                        help='number of gpus per node (default: 1)')
+    parser.add_argument('-n', '--nodes', default=1, type=int, metavar='N',
+                        help='number of machines (default: 1)')
     parser.add_argument('-nr', '--nr', default=0, type=int,
-                        help='ranking within the nodes')
-    parser.add_argument('--epochs', default=2, type=int, metavar='N',
-                        help='number of total epochs to run')
+                        help='ranking within the nodes (default: 0)')
+    parser.add_argument('-e', '--epochs', default=2, type=int, metavar='N',
+                        help='number of total epochs to run (default: 2)')
     args = parser.parse_args()
     train(0, args)
 

multi_gpu/main.py

@@ -65,14 +65,14 @@ def train(gpu, args):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument('-n', '--nodes', default=1, type=int, metavar='N',
-                        help='number of data loading workers (default: 4)')
     parser.add_argument('-g', '--gpus', default=1, type=int,
-                        help='number of gpus per node')
+                        help='number of gpus per node (default: 1)')
+    parser.add_argument('-n', '--nodes', default=1, type=int, metavar='N',
+                        help='number of machines (default: 1)')
     parser.add_argument('-nr', '--nr', default=0, type=int,
-                        help='ranking within the nodes')
-    parser.add_argument('--epochs', default=2, type=int, metavar='N',
-                        help='number of total epochs to run')
+                        help='ranking within the nodes (default: 0)')
+    parser.add_argument('-e', '--epochs', default=2, type=int, metavar='N',
+                        help='number of total epochs to run (default: 2)')
     args = parser.parse_args()
     # train(0, args)
     args.world_size = args.gpus * args.nodes

single_gpu/main.py

@@ -51,14 +51,8 @@ def train(gpu, args):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument('-n', '--nodes', default=1, type=int, metavar='N',
-                        help='number of data loading workers (default: 4)')
-    parser.add_argument('-g', '--gpus', default=1, type=int,
-                        help='number of gpus per node')
-    parser.add_argument('-nr', '--nr', default=0, type=int,
-                        help='ranking within the nodes')
-    parser.add_argument('--epochs', default=2, type=int, metavar='N',
-                        help='number of total epochs to run')
+    parser.add_argument('-e', '--epochs', default=2, type=int, metavar='N',
+                        help='number of total epochs to run (default: 2)')
     args = parser.parse_args()
     train(0, args)