@classmethod
def setUpClass(cls):
    if (not torch.cuda.is_available()):
        raise unittest.SkipTest('No CUDA capable device detected')
@classmethod
def setUpClass(cls):
    if (not torch.cuda.is_available()):
        raise unittest.SkipTest('No CUDA capable device detected')