Fix max_steps

This commit is contained in:
Matthw Honnibal 2020-07-01 18:08:14 +02:00
parent cb51bb637b
commit 6a0a27e5c2
1 changed files with 1 additions and 1 deletions

View File

@ -462,7 +462,7 @@ def train_while_improving(
if patience and (step - best_step) >= patience:
break
# Stop if we've exhausted our max steps (if specified)
if max_steps and (step * accumulate_gradient) >= max_steps:
if max_steps and step >= max_steps:
break