data

Functions

extract_batch_size	Unpack a batch to find a torch.Tensor.
has_len_all_ranks	Checks if a given object has __len__ method implemented on all ranks.

lightning.pytorch.utilities.data.extract_batch_size(batch)

Unpack a batch to find a torch.Tensor.

Return type:

int

Returns:

len(tensor) when found, or 1 when it hits an empty or non iterable.

lightning.pytorch.utilities.data.has_len_all_ranks(dataloader, strategy, allow_zero_length_dataloader_with_multiple_devices=False)

Checks if a given object has __len__ method implemented on all ranks.

Return type:

TypeGuard[Sized]