File size: 442 Bytes
ce209f5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | 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"))
|