Jingqiao-ucsc's picture
Upload WiSER private archive chunk: code_snapshots
13a1c10 verified
Raw
History Blame Contribute Delete
722 Bytes
"""WiSER: a wireless scene encoder for radiomap and CIR prediction.
The public API uses CIR terminology. Some internal state-dict keys retain the
historical ``csi_*`` prefix so that the released checkpoint remains loadable.
"""
from .data.dataset import TriplePathDataset
from .engine.trainer import CsiSetTrainer
from .models.model import SparseCsiDetrModel
from .alt_models import FreezeConfig, JointRadiomapCIRModel
__version__ = "0.0.1.dev0"
CIRSetTrainer = CsiSetTrainer
SparseCIRDetrModel = SparseCsiDetrModel
__all__ = [
"__version__",
"CIRSetTrainer",
"FreezeConfig",
"JointRadiomapCIRModel",
"CsiSetTrainer",
"SparseCIRDetrModel",
"SparseCsiDetrModel",
"TriplePathDataset",
]