from __future__ import annotations from dataloader import VCDADataset # noqa: F401 from loss_fn import setup_criterion # noqa: F401 from model import ResNet18Head # noqa: F401