Merge pull request #35 from boostcampaitech7/ljb-issue-08

Feat: Dense Retrieval 구현
boostcampaitech7 · Oct 16, 2024 · 1801c2e · 1801c2e
2 parents bfa40f8 + 6e85ac5
commit 1801c2e
Show file tree

Hide file tree

Showing 3 changed files with 578 additions and 5 deletions.
diff --git a/config.yaml b/config.yaml
@@ -14,6 +14,7 @@ training:
   scheduler: linear # HF default = "linear"
 
 dataRetrieval:
+  type: sparse
   eval: True
   context_path: ./data/wikipedia_documents.json
   top_k: 10