File size: 426 Bytes
0ab2566 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """
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()
|