def to_device()

in utils/torch_helpers.py [0:0]

• 11 lines of code
• 6 McCabe index (conditional complexity)

def to_device(data):
    if isinstance(data, torch.Tensor):
        data = data.to(_device, non_blocking=True)
        return data

    if isinstance(data, dict):
        for k, v in data.items():
            data[k] = to_device(v)
        return data

    # list or tuple
    for i, v in enumerate(data):
        data[i] = to_device(v)
    return data