rename field

This commit is contained in:
svlandeg 2020-08-28 14:06:23 +02:00
parent 1d8c4070aa
commit 33883aa764
1 changed files with 2 additions and 2 deletions

View File

@ -67,7 +67,7 @@ def console_logger():
@registry.loggers("spacy.WandbLogger.v1") @registry.loggers("spacy.WandbLogger.v1")
def wandb_logger(project_name: str, disable_fields: list = []): def wandb_logger(project_name: str, remove_config_values: list = []):
import wandb import wandb
console = console_logger() console = console_logger()
@ -77,7 +77,7 @@ def wandb_logger(project_name: str, disable_fields: list = []):
) -> Tuple[Callable[[Dict[str, Any]], None], Callable]: ) -> Tuple[Callable[[Dict[str, Any]], None], Callable]:
config = nlp.config.interpolate() config = nlp.config.interpolate()
config_dot = util.dict_to_dot(config) config_dot = util.dict_to_dot(config)
for field in disable_fields: 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)