Merge pull request #1 from tianxin1860/develop

stop training by num_train_steps

Merge pull request #1 from tianxin1860/develop
stop training by num_train_steps
f5c0ec27 · Yibing Liu · GitHub · c54d5f93 · 8ff463b7 · f5c0ec27
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

BERT/train.py BERT/train.py +1 -1

未找到文件。
--- a/BERT/train.py
+++ b/BERT/train.py
@@ -335,7 +335,7 @@ def train(args):
    lm_cost = []
    acc = []
    time_begin = time.time()
-    while True:
+    while steps < args.num_train_steps:
        try:
            steps += nccl2_num_trainers
            skip_steps = args.skip_steps * nccl2_num_trainers