Skip to content

GitLab

"git@developer.sourcefind.cn:lacacy/qwen_lmdeploy.git" did not exist on "79deb99dbc4417ae1619b3807bd993555910bec8"
Commit be9c80a4 authored by dcuai's avatar dcuai
Browse files

Update main_train_psnr.py '--local-rank'

parent e49ba488
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
main_train_psnr.py
View file @ be9c80a4
...@@ -51,7 +51,7 @@ def main(json_path='options/masked_denoising/input_mask_80_90.json'): ...@@ -51,7 +51,7 @@ def main(json_path='options/masked_denoising/input_mask_80_90.json'):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--opt', type=str, default=json_path, help='Path to option JSON file.') parser.add_argument('--opt', type=str, default=json_path, help='Path to option JSON file.')
parser.add_argument('--launcher', default='pytorch', help='job launcher') parser.add_argument('--launcher', default='pytorch', help='job launcher')
parser.add_argument('--local_rank', type=int, default=0) parser.add_argument('--local-rank', type=int, default=0)
parser.add_argument('--epochs', type=int, default=1000000) parser.add_argument('--epochs', type=int, default=1000000)
parser.add_argument('--dist', default=False) parser.add_argument('--dist', default=False)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment