Batch Manager#

Module for the Batch Manager class.

class _BatchManager[source]#

Bases: dict

Dict-like container for batched data with attribute-style access and convenience methods for device placement.

to(device)[source]#

Move all compatible values in the batch to the specified device.

Parameters:

device (torch.device | str) – The target device.

Returns:

The updated batch manager.

Return type:

_BatchManager