enable_cudnn_benchmark

dragon.cuda.enable_cudnn_benchmark(enabled=True)[source]

Activate the CuDNN benchmark.

Parameters:
  • enabled (bool, optional, default=True) – True to activate CuDNN benchmark.