Types & classes — github.com/LUMIA-Group/MemoryDecoder

CodeHub MCPcopy Create free account

hub / github.com/LUMIA-Group/MemoryDecoder / types & classes

Types & classes14 in github.com/LUMIA-Group/MemoryDecoder

⨍ Functions 73 ◇ Types & classes 14

↓ 2 callersClassActivationCapturer

knn_utils/saveEmbedMulti.py:479

↓ 2 callersClassInterceptHandler

evaluate_joint.py:316

↓ 2 callersClassInterceptHandler

train_base.py:228

↓ 2 callersClassInterceptHandler

train_memdec.py:356

↓ 1 callersClassKNNSaverMulti

knn_utils/saveEmbedMulti.py:284

↓ 1 callersClassKNNSearchMulti

knn_utils/saveKNNMulti.py:20

↓ 1 callersClassKNNWrapperMulti

knn_utils/saveEmbedMulti.py:47

↓ 1 callersClassMemoryDecoder

A light wrapper around **two** causal‑LMs that fuses their logits: logits_joint = logaddexp(logits_base + log(1‑λ),

demo/memDec.py:30

↓ 1 callersClassMemoryDecoderOutput

demo/memDec.py:21

knn_utils/saveEmbedMulti.py:25

ClassDataTrainingArguments

Arguments pertaining to what data we are going to input our model for training and eval.

train_base.py:138

knn_utils/saveEmbedMulti.py:36

ClassKNNArguments

Arguments pertaining to which model/config/tokenizer we are going to fine-tune, or train from scratch.

train_base.py:188

ClassModelArguments

Arguments pertaining to which model/config/tokenizer we are going to fine-tune, or train from scratch.

train_base.py:74