train/train_dsamnet.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_torchvision_weight
from wrapper_common import ROOT, main_for


DSAMNET_ROOT = ROOT / "model_repos" / "DSAMNet"
if DSAMNET_ROOT.exists():
    sys.path.insert(0, str(DSAMNET_ROOT))


def ensure_dsamnet_backbone() -> None:
    ensure_torchvision_weight("resnet18")


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