relik-ie commited on
Commit
e888405
1 Parent(s): be7b747

Create config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +17 -0
config.yaml ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_: relik.inference.annotator.Relik
2
+ retriever:
3
+ triplet:
4
+ _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
5
+ question_encoder: relik-ie/encoder-e5-small-v2-wikipedia-relations
6
+ index:
7
+ triplet:
8
+ _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
9
+ name_or_path: relik-ie/encoder-e5-small-v2-wikipedia-relations-index
10
+ reader:
11
+ _target_: relik.reader.pytorch_modules.triplet.RelikReaderForTripletExtraction
12
+ transformer_model: relik-ie/relik-reader-deberta-v3-large-re-wikipedia
13
+ task: TRIPLET
14
+ metadata_fields: []
15
+ top_k: 30
16
+ window_size: 64
17
+ window_stride: 32