torchsig.datasets.datamodulesΒΆ

PyTorch Lightning DataModules for Narrowband and Wideband

Learn More: https://lightning.ai/docs/pytorch/stable/data/datamodule.html

If dataset does not exist at root, creates new dataset and writes to disk If dataset does exsit, simply loaded it back in

Classes

NarrowbandDataModule

DataModule for creating and managing narrowband datasets.

OfficialNarrowbandDataModule

A DataModule for the official Narrowband dataset.

OfficialTorchSigDataModdule

A PyTorch Lightning DataModule for official TorchSignal datasets.

OfficialWidebandDataModule

A DataModule for the official Wideband dataset.

TorchSigDataModule

PyTorch Lightning DataModule for managing TorchSig datasets.

WidebandDataModule

DataModule for creating and managing wideband datasets.