liqi / code /pmps /train_direct_match.py
WaitXian's picture
Upload liqi files
0ab2566 verified
Raw History Blame Contribute Delete
426 Bytes
"""
Ablation: Direct Soft Matching (no prototypes).
Reuses the full training pipeline from train.py, only swapping
CODI_PMPS (prototype-mediated) with CODI_DirectMatch (direct cosine matching).
"""
import train as train_module
from src.model_direct_match import CODI_DirectMatch
# Replace the model class before invoking train()
train_module.CODI_PMPS = CODI_DirectMatch
if __name__ == "__main__":
train_module.train()