def test_linear_model_multi_gpu(self): with cuda.Device(0): self.assertGreater( cuda.to_cpu(self.model.accuracy_gpu(1).data), 0.9)