| _target_: src.pipeline.NerRePipeline | |
| ner_model_path: ??? | |
| re_model_path: ??? | |
| entity_layer: labeled_spans | |
| relation_layer: binary_relations | |
| # this works good on a single RTX2080Ti (11GB) | |
| ner_pipeline: | |
| batch_size: 256 | |
| taskmodule_kwargs: | |
| # to get more correct scores (default in the taskmodule is "mean") | |
| combine_token_scores_method: product | |
| re_pipeline: | |
| batch_size: 64 | |
| # convert the RE model to half precision for mixed precision inference (speedup approx. 4x) | |
| half_precision_model: true | |
| taskmodule_kwargs: | |
| # don't show statistics after encoding | |
| collect_statistics: false | |
| # don't show pipeline steps | |
| verbose: false | |