Download code/pmps/train_direct_match.py from WaitXian/liqi: direct link, hf CLI and curl.
- Browser
- Download file 426 Bytes
-
https://huggingface.co/WaitXian/liqi/resolve/main/code/pmps/train_direct_match.py
- Command line
-
hf download hf://WaitXian/liqi/code/pmps/train_direct_match.py
-
curl -L -o train_direct_match.py https://huggingface.co/WaitXian/liqi/resolve/main/code/pmps/train_direct_match.py
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() | |