src.utils.data_to_cuda.data_to_cuda

src.utils.data_to_cuda.data_to_cuda(inputs)

Call cuda() on all tensor elements in inputs :param inputs: input list/dictionary :return: identical to inputs while all its elements are on cuda