myoverse.datasets.datamodule.DataModule.__init__#
- DataModule.__init__(data_path, inputs=('emg',), targets=('kinematics',), batch_size=32, window_size=200, window_stride=None, n_windows_per_epoch=None, num_workers=4, train_transform=None, val_transform=None, test_transform=None, target_transform=None, pin_memory=True, persistent_workers=True, device=None, dtype=torch.float32, cache_in_ram=True)[source]#
- Parameters:
batch_size (int)
window_size (int)
window_stride (int | None)
n_windows_per_epoch (int | None)
num_workers (int)
train_transform (Callable | None)
val_transform (Callable | None)
test_transform (Callable | None)
target_transform (Callable | None)
pin_memory (bool)
persistent_workers (bool)
dtype (dtype)
cache_in_ram (bool)
- prepare_data_per_node#
If True, each LOCAL_RANK=0 will call prepare data. Otherwise only NODE_RANK=0, LOCAL_RANK=0 will prepare data.
- myoverse.datasets.datamodule.allow_zero_length_dataloader_with_multiple_devices#
If True, dataloader with zero length within local rank is allowed. Default value is False.