train/train_tinycd.py

from __future__ import annotations

import sys
from pathlib import Path

LOCAL_ROOT = Path(__file__).resolve().parents[1]
if str(LOCAL_ROOT) not in sys.path:
    sys.path.insert(0, str(LOCAL_ROOT))

from utils.weight_downloader import ensure_timm_weight
from wrapper_common import main_for


def ensure_tinycd_backbone() -> None:
    ensure_timm_weight("efficientnet_b4")


if __name__ == "__main__":
    raise SystemExit(main_for("tinycd"))