Download 1.3B/code/tests/test_triple_partial_geometry.py from Elastic-forcing/Elastic-forcing-code: direct link, hf CLI and curl.
- Browser
- Download file 576 Bytes
-
https://huggingface.co/Elastic-forcing/Elastic-forcing-code/resolve/main/1.3B/code/tests/test_triple_partial_geometry.py
- Command line
-
hf download hf://Elastic-forcing/Elastic-forcing-code/1.3B/code/tests/test_triple_partial_geometry.py
-
curl -L -o test_triple_partial_geometry.py https://huggingface.co/Elastic-forcing/Elastic-forcing-code/resolve/main/1.3B/code/tests/test_triple_partial_geometry.py
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] | |