chainer.backends.cuda.set_max_workspace_size

chainer.backends.cuda.set_max_workspace_size(size)[source]

Sets the workspace size for cuDNN.

Check “cuDNN Library User Guide” for detail.

Parameters:size – The workspace size for cuDNN.