Merge pull request #289 from yuedongli1/master

delete device_per_servers, only used for log
mindspore-lab · Apr 29, 2024 · f8b3df2 · f8b3df2
2 parents 6737cec + b97ab64
commit f8b3df2
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -23,7 +23,6 @@ def get_parser_train(parents=None):
     parser.add_argument("--task", type=str, default="detect", choices=["detect", "segment"])
     parser.add_argument("--device_target", type=str, default="Ascend", help="device target, Ascend/GPU/CPU")
     parser.add_argument("--save_dir", type=str, default="./runs", help="save dir")
-    parser.add_argument("--device_per_servers", type=int, default=8, help="device number on a server")
     parser.add_argument("--log_level", type=str, default="INFO", help="log level to print")
     parser.add_argument("--is_parallel", type=ast.literal_eval, default=False, help="Distribute train or not")
     parser.add_argument("--ms_mode", type=int, default=0,