Elastic-forcing-code / 1.3B /code /tests /test_triple_partial_geometry.py
Elastic-forcing's picture
Add training and inference source code
2ea2835 verified
Raw History Blame Contribute Delete
576 Bytes
from pathlib import Path
def test_released_recipe_uses_10_1_1_loss_divided_by_three():
import yaml
from utils.config import resolve_config_aliases
root = Path(__file__).resolve().parents[2]
config = resolve_config_aliases(yaml.safe_load((root / "configs/three_encoder.yaml").read_text()))
assert config["triple_dinov3_weight"] == 10.0
assert config["triple_vjepa2_weight"] == 1.0
assert config["triple_videomae_v2_weight"] == 1.0
assert config["triple_aggregation_divisor"] == 3.0
assert config["partial_mmd_geometry"] == [256, 128, 128]