Method test_transfer_learning_gpu

test/python/test_onnx.py:2123–2124

(self)

    @unittest.skipIf(not singa_api.USE_CUDA, 'CUDA is not enabled')
    def test_transfer_learning_gpu(self):
        self._transfer_learning_helper(gpu_dev)

if __name__ == '__main__':
    unittest.main()