corrected on call to cuda in tests, now called only when USE_GPU=True
@@ -107,6 +107,7 @@ class ConversionMethodTests(unittest.TestCase):
module = module.eval()
self.assertIsInstance(module, MultiTaskModule)
self.assertFalse(module.training)
+ if self.device == 'cuda':
module = module.cuda()
self.assertIsInstance(module, MultiTaskModule)
