src.utils.data_to_cudaΒΆ

Functions

data_to_cuda

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