Skip to content

GitLab

Unverified Commit 8ac072ae authored by J-shang's avatar J-shang Committed by GitHub
Browse files

[Bugbash] promote pruning v2 (#4733)

parent 1a3c019a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/ut/compression/v2/test_pruning_wrapper.py
View file @ 8ac072ae
...@@ -6,7 +6,7 @@ import unittest ...@@ -6,7 +6,7 @@ import unittest
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
from nni.algorithms.compression.v2.pytorch.pruning import L1NormPruner from nni.compression.pytorch.pruning import L1NormPruner
class TorchModel(torch.nn.Module): class TorchModel(torch.nn.Module):
def __init__(self): def __init__(self):
......
test/ut/compression/v2/test_scheduler.py
View file @ 8ac072ae
...@@ -6,7 +6,7 @@ import unittest ...@@ -6,7 +6,7 @@ import unittest
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
from nni.algorithms.compression.v2.pytorch.pruning import PruningScheduler, L1NormPruner from nni.compression.pytorch.pruning import PruningScheduler, L1NormPruner
from nni.algorithms.compression.v2.pytorch.pruning.tools import AGPTaskGenerator from nni.algorithms.compression.v2.pytorch.pruning.tools import AGPTaskGenerator
......
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