wsl-base / config.yaml
andreim14's picture
added config and readme
6b95e98
raw
history blame
559 Bytes
_target_: wsl.inference.annotator.WSL
retriever:
span:
_target_: wsl.retriever.pytorch_modules.model.WSLRetriever
question_encoder: Babelscape/wsl-retriever-e5-base-v2
index:
span:
_target_: wsl.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
name_or_path: Babelscape/wsl-retriever-e5-base-v2-worndnet-index
reader:
_target_: wsl.reader.pytorch_modules.span.WSLReaderForSpanExtraction
transformer_model: Babelscape/wsl-reader-deberta-v3-base
task: SPAN
metadata_fields: []
top_k: 100
window_size: 16
window_stride: 8