Skip to content

GitLab

Unverified Commit 29b4d46c authored by John's avatar John Committed by GitHub
Browse files

Update amc_pruner.py (#4161)

parent 5f0a7c9a
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
nni/algorithms/compression/pytorch/pruning/amc/amc_pruner.py
View file @ 29b4d46c
...@@ -14,8 +14,6 @@ from .channel_pruning_env import ChannelPruningEnv ...@@ -14,8 +14,6 @@ from .channel_pruning_env import ChannelPruningEnv
from .lib.agent import DDPG from .lib.agent import DDPG
from .lib.utils import get_output_folder from .lib.utils import get_output_folder
torch.backends.cudnn.deterministic = True
_logger = logging.getLogger(__name__) _logger = logging.getLogger(__name__)
class AMCPruner(Pruner): class AMCPruner(Pruner):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment