[examples] consistently use --gpus, instead of --n_gpu (#6315)

2020-08-07 07:36:32 -07:00
parent 0e36e51515
commit 6695450a23
4 changed files with 5 additions and 9 deletions
--- a/examples/distillation/train.py
+++ b/examples/distillation/train.py
@@ -210,7 +210,7 @@ def main():
        help="For fp16: Apex AMP optimization level selected in ['O0', 'O1', 'O2', and 'O3']."
        "See details at https://nvidia.github.io/apex/amp.html",
    )
-    parser.add_argument("--n_gpu", type=int, default=1, help="Number of GPUs in the node.")
+    parser.add_argument("--gpus", type=int, default=1, help="Number of GPUs in the node.")
    parser.add_argument("--local_rank", type=int, default=-1, help="Distributed training - Local rank")
    parser.add_argument("--seed", type=int, default=56, help="Random seed")