Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix wandb logger when calling multiple times from same script

This commit is contained in:
svlandeg 2020-09-15 12:56:33 +02:00
parent b7faa38960
commit 7677e5c0e2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/training/loggers.py
View File

@ -72,7 +72,7 @@ def wandb_logger(project_name: str, remove_config_values: List[str] = []):
for field in remove_config_values: for field in remove_config_values:
del config_dot[field] del config_dot[field]
config = util.dot_to_dict(config_dot) config = util.dot_to_dict(config_dot)
wandb.init(project=project_name, config=config) wandb.init(project=project_name, config=config, reinit=True)
console_log_step, console_finalize = console(nlp) console_log_step, console_finalize = console(nlp)
def log_step(info: Dict[str, Any]): def log_step(info: Dict[str, Any]):
@ -88,7 +88,7 @@ def wandb_logger(project_name: str, remove_config_values: List[str] = []):
def finalize(): def finalize():
console_finalize() console_finalize()
pass wandb.join()
return log_step, finalize return log_step, finalize