File size: 475 Bytes
8b0b874 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | #!/usr/bin/env python3
"""Run the shared SiT trainer with the rotation-layer model implementation."""
import os
import runpy
# train.py imports its model registry at module load time. Set both variables
# first, and require the shared trainer to fail closed if another registry is
# ever selected accidentally.
os.environ["SIT_MODEL_MODULE"] = "models_rot_layer"
os.environ["SIT_EXPECTED_MODEL_MODULE"] = "models_rot_layer"
runpy.run_module("train", run_name="__main__")
|