explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-12 02:06:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow output_path to be None during training

Browse Source
This commit is contained in:
Ines Montani 2021-01-25 11:51:19 +11:00
parent ffc371350a
commit 146af1dc17
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/training/loop.py
View File

@ -100,7 +100,8 @@ def train(
if is_best_checkpoint is not None and output_path is not None:
with nlp.select_pipes(disable=frozen_components):
update_meta(T, nlp, info)
save_checkpoint(is_best_checkpoint)
if output_path is not None:
save_checkpoint(is_best_checkpoint)
except Exception as e:
if output_path is not None:
stdout.write(
@ -113,7 +114,8 @@ def train(
raise e
finally:
finalize_logger()
save_checkpoint(False)
if output_path is not None:
save_checkpoint(False)
# This will only run if we did't hit an error
if optimizer.averages:
nlp.use_params(optimizer.averages)