Fix logic in train CLI timing eval on GPU (#5387)

Run CPU timing in first iteration only
2020-05-01 12:05:33 +02:00 · 2020-05-01 12:05:33 +02:00 · c045a9c7f6
parent 148b036e0c
commit c045a9c7f6
1 changed files with 1 additions and 1 deletions
--- a/spacy/cli/train.py
+++ b/spacy/cli/train.py
@ -456,7 +456,7 @@ def train(
                        gpu_wps = nwords / (end_time - start_time)
                        # Only evaluate on CPU in the first iteration (for
                        # timing) if GPU is enabled
-                        if i >= 1:
+                        if i == 0:
                            with Model.use_device("cpu"):
                                nlp_loaded = util.load_model_from_path(epoch_model_path)
                                for name, component in nlp_loaded.pipeline: