From bc72b3c9bd99cccf528a1b938207c008e3cfc92a Mon Sep 17 00:00:00 2001 From: ydlme Date: Tue, 31 Oct 2023 21:44:32 +0800 Subject: [PATCH] Fix run finetune.py from torch.distributed.launch --- finetune/finetune.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/finetune/finetune.py b/finetune/finetune.py index 96ab961..9818bca 100644 --- a/finetune/finetune.py +++ b/finetune/finetune.py @@ -76,7 +76,7 @@ def get_args(): parser.add_argument("--num_warmup_steps", type=int, default=100) parser.add_argument("--weight_decay", type=float, default=0.05) - parser.add_argument("--local_rank", type=int, default=0) + parser.add_argument("--local-rank", type=int, default=0) parser.add_argument("--no_fp16", action="store_false") parser.add_argument("--bf16", action="store_true", default=True) parser.add_argument("--no_gradient_checkpointing", action="store_false", default=False) @@ -309,4 +309,4 @@ def main(args): logging.set_verbosity_error() - main(args) \ No newline at end of file + main(args)