summaryrefslogtreecommitdiff
path: root/configs/default.yaml
diff options
context:
space:
mode:
authorYurenHao0426 <blackhao0426@gmail.com>2026-02-15 18:19:50 +0000
committerYurenHao0426 <blackhao0426@gmail.com>2026-02-15 18:19:50 +0000
commitc90b48e3f8da9dd0f8d2ae82ddf977436bb0cfc3 (patch)
tree43edac8013fec4e65a0b9cddec5314489b4aafc2 /configs/default.yaml
Initial implementation of HAG (Hopfield-Augmented Generation)HEADmaster
Core Hopfield retrieval module with energy-based convergence guarantees, memory bank, FAISS baseline retriever, evaluation metrics, and end-to-end pipeline. All 45 tests passing on CPU with synthetic data. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Diffstat (limited to 'configs/default.yaml')
-rw-r--r--configs/default.yaml21
1 files changed, 21 insertions, 0 deletions
diff --git a/configs/default.yaml b/configs/default.yaml
new file mode 100644
index 0000000..cb76d34
--- /dev/null
+++ b/configs/default.yaml
@@ -0,0 +1,21 @@
+hopfield:
+ beta: 1.0
+ max_iter: 5
+ conv_threshold: 1.0e-4
+ top_k: 5
+
+memory:
+ embedding_dim: 768
+ normalize: true
+
+encoder:
+ model_name: "facebook/contriever-msmarco"
+ max_length: 512
+ batch_size: 64
+
+generator:
+ model_name: "meta-llama/Llama-3.1-8B-Instruct"
+ max_new_tokens: 128
+ temperature: 0.0
+
+retriever_type: "hopfield"