def batch_to_cuda()

in depth_upsampling/data_utils.py [0:0]

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

def batch_to_cuda(batch):
    if torch.cuda.is_available():
        for k in batch:
            if k != 'identifier':
                batch[k] = batch[k].cuda(non_blocking=True)
    return batch