Update amc_pruner.py (#4161)

microsoft · Sep 8, 2021 · 29b4d46 · 29b4d46
1 parent 5f0a7c9
commit 29b4d46
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/nni/algorithms/compression/pytorch/pruning/amc/amc_pruner.py b/nni/algorithms/compression/pytorch/pruning/amc/amc_pruner.py
@@ -14,8 +14,6 @@
 from .lib.agent import DDPG
 from .lib.utils import get_output_folder
 
-torch.backends.cudnn.deterministic = True
-
 _logger = logging.getLogger(__name__)
 
 class AMCPruner(Pruner):