relik-cie-small / config.yaml
riccorl's picture
Automatic push from sapienzanlp
5cac46f
_target_: relik.inference.annotator.Relik
retriever:
span:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: riccorl/retriever-relik-cie-small-span-question-encoder
document_index: riccorl/retriever-relik-cie-small-span-index
triplet:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: riccorl/retriever-relik-cie-small-triplet-question-encoder
document_index: riccorl/retriever-relik-cie-small-triplet-index
reader:
_target_: relik.reader.pytorch_modules.triplet.RelikReaderForTripletExtraction
transformer_model: riccorl/reader-relik-cie-small
task: BOTH
metadata_fields: []
top_k: 100
window_size: sentence
window_stride: null