def _batch2device(
self, batch: Mapping[str, Any], device: Device,
) -> Mapping[str, Any]:
"""
Inner method to transfer incoming data batches to Runners' device.
Args:
batch (Mapping[str, Any]): dictionary with data batches
from DataLoader.
device (Device): torch device
Returns:
Mapping[str, Any]: same structure as value,
but all tensors and np.arrays moved to device
"""
output = utils.any2device(batch, device)
return output
def _batch2device(
self,
batch: Mapping[str, Any],
device: Device,
) -> Mapping[str, Any]:
"""
Inner method to transfer incoming data batches to Runners' device.
Args:
batch (Mapping[str, Any]): dictionary with data batches
from DataLoader.
device (Device): torch device
Returns:
"""
output = utils.any2device(batch, device)
return output