zhreshold commented on a change in pull request #12691: [MXNET-1001] Add new
cuddn flag in Gluon Conv blocks to control - cudnn_tune, cudnn_off params
URL: https://github.com/apache/incubator-mxnet/pull/12691#discussion_r227076788
##########
File path: python/mxnet/gluon/nn/conv_layers.py
##########
@@ -86,10 +121,20 @@ class _Conv(HybridBlock):
Initializer for the `weight` weights matrix.
bias_initializer: str or `Initializer`
Initializer for the bias vector.
+ cudnn : str,
+ Supported values - {'off', 'default', 'fastest', 'limited_workspace',
'global'}.
Review comment:
by default it should align with env variable, i.e., the global in your
revision. And I don't quite like the name global, sounds like a global tuning
option. Is env a better name?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
