sbodenstein edited a comment on issue #12691: [MXNET-1001] [WIP] Add new cudnn_tune, cudnn_off params in gluon conv blocks URL: https://github.com/apache/incubator-mxnet/pull/12691#issuecomment-427326365 @szha: I do agree in general. But with cuDNN, you sometimes want to disable it only for specific layers. For example, we disable cuDNN for batchnorm layers when `eps < 10^-5`, as cuDNN fails in this case (and sometimes you import nets from other frameworks with smaller values for `eps` that you want to run). You definitely don't want a global variable switching off cuDNN for all layers in the net. The ideal would probably be to have better fallbacks in this case though rather than an extra flag...
---------------------------------------------------------------- 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
