ksuit.utils.torch.device

Functions

move_items_to_device(device, batch)

Moves everything in the batch to the given device.

Module Contents

ksuit.utils.torch.device.move_items_to_device(device, batch)

Moves everything in the batch to the given device.

Parameters:

batch (dict[str, Any])