From b97ab643caa7195c447c52a5592a9e2eb0ce5f7a Mon Sep 17 00:00:00 2001 From: yuedongli1 Date: Thu, 25 Apr 2024 10:06:33 +0800 Subject: [PATCH] delete device_per_servers, only used for log --- train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train.py b/train.py index ddb5d49f..cecce995 100644 --- 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,