Skip to content

Commit 4bfedf1

Browse files
committed
Allow the pipeline to pass along a --coref_log_norms parameter to the coref model
1 parent 96f9c2c commit 4bfedf1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stanza/pipeline/coref_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def _set_up_model(self, config, pipeline, device):
7070
# (except its config)
7171
# TODO: separate any pretrains if possible
7272
# TODO: add device parameter to the load mechanism
73-
config_update = {'log_norms': False,
73+
config_update = {'log_norms': config.get('log_norms', False),
7474
'device': device}
7575
model = CorefModel.load_model(path=config['model_path'],
7676
ignore={"bert_optimizer", "general_optimizer",

0 commit comments

Comments
 (0)