Skip to content

GitLab

Commit 7a3a7c53 authored by dcuai's avatar dcuai
Browse files

Update train.py

parent f197d0a5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train.py
View file @ 7a3a7c53
...@@ -278,7 +278,7 @@ def main(): ...@@ -278,7 +278,7 @@ def main():
parser.add_argument("--max_grad_norm", default=1.0, type=float, parser.add_argument("--max_grad_norm", default=1.0, type=float,
help="Max gradient norm.") help="Max gradient norm.")
parser.add_argument("--local_rank", type=int, default=-1, parser.add_argument("--local-rank", type=int, default=-1,
help="local_rank for distributed training on gpus") help="local_rank for distributed training on gpus")
parser.add_argument('--seed', type=int, default=42, parser.add_argument('--seed', type=int, default=42,
help="random seed for initialization") help="random seed for initialization")
......
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