Python chainer.cuda 模块,get_max_workspace_size() 实例源码

我们从Python开源项目中,提取了以下2个代码示例,用于说明如何使用chainer.cuda.get_max_workspace_size()

项目:chainer-deconv    作者:germanRos    | 项目源码 | 文件源码
def setUp(self):
        self.space = cuda.get_max_workspace_size()
项目:chainer-deconv    作者:germanRos    | 项目源码 | 文件源码
def test_size(self):
        size = 1024
        cuda.set_max_workspace_size(size)
        self.assertEqual(size, cuda.get_max_workspace_size())