UnlabeledDataModule

class lightning_pose.data.datamodules.UnlabeledDataModule(dataset: Dataset, video_paths_list: List[str] | str, dali_config: dict | DictConfig, view_names: List[str] | None = None, train_batch_size: int = 16, val_batch_size: int = 16, test_batch_size: int = 1, num_workers: int = 8, train_probability: float = 0.8, val_probability: float | None = None, test_probability: float | None = None, train_frames: float | None = None, torch_seed: int = 42, imgaug: Literal['default', 'dlc', 'dlc-top-down'] = 'default')

Bases: BaseDataModule

Data module that contains labeled and unlabled data loaders.

Methods Summary

setup_unlabeled()	Sets up the unlabeled data loader.
train_dataloader()	An iterable or collection of iterables specifying training samples.

Methods Documentation

setup_unlabeled() → None

Sets up the unlabeled data loader.

train_dataloader() → CombinedLoader

An iterable or collection of iterables specifying training samples.

For more information about multiple dataloaders, see this section.

The dataloader you return will not be reloaded unless you set :paramref:`~lightning.pytorch.trainer.trainer.Trainer.reload_dataloaders_every_n_epochs` to a positive integer.

For data processing use the following pattern:

• download in prepare_data()

• process and split in setup()

However, the above are only necessary for distributed processing.

Warning

do not assign state in prepare_data

• fit()

• prepare_data()

• setup()

Note

Lightning tries to add the correct sampler for distributed and arbitrary hardware. There is no need to set it yourself.