| Server IP : 217.154.3.148 / Your IP : 216.73.216.90 Web Server : nginx/1.30.3 System : Linux blissful-wright.217-154-3-148.plesk.page 6.8.0-124-generic #124-Ubuntu SMP PREEMPT_DYNAMIC Tue May 26 13:00:45 UTC 2026 x86_64 User : gracious-boyd_hfhh1h8vo9n ( 10000) PHP Version : 8.3.31 Disable Function : opcache_get_status MySQL : OFF | cURL : ON | WGET : OFF | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /var/www/vhosts/gracious-boyd.217-154-3-148.plesk.page/nextjs/output/ |
Upload File : |
"""Wrapper module so runtime scripts can import the trained Lightning module.
The actual training implementation lives under models/unet_latest/train_unet.py
which defines the LitSeg LightningModule as well as the auxiliary dataset and
loss classes. Exposing them from this module keeps the command line tools stable
(they can simply use --lit-module output.train_unet) while allowing us to move
around the training sources.
"""
from models.unet_latest.train_unet import (
DiceLoss,
LitSeg,
SegDataset,
UNet,
)
__all__ = [
"DiceLoss",
"LitSeg",
"SegDataset",
"UNet",
]