Unverified Commit 2fc3f0b3 authored by chenbohua3's avatar chenbohua3 Committed by GitHub
Browse files

no necessary to use pretrained model in pruner ut (#4085)

parent 6d9c49da
...@@ -79,7 +79,7 @@ class DependencyawareTest(TestCase): ...@@ -79,7 +79,7 @@ class DependencyawareTest(TestCase):
print('Testing on ', pruner) print('Testing on ', pruner)
ori_filters = {} ori_filters = {}
Model = getattr(models, model_name) Model = getattr(models, model_name)
net = Model(pretrained=True, progress=False) net = Model(pretrained=False, progress=False)
# record the number of the filter of each conv layer # record the number of the filter of each conv layer
for name, module in net.named_modules(): for name, module in net.named_modules():
if isinstance(module, nn.Conv2d): if isinstance(module, nn.Conv2d):
...@@ -127,7 +127,7 @@ class DependencyawareTest(TestCase): ...@@ -127,7 +127,7 @@ class DependencyawareTest(TestCase):
Model = getattr(models, model_name) Model = getattr(models, model_name)
cfg_generator = [generate_random_sparsity, generate_random_sparsity_v2] cfg_generator = [generate_random_sparsity, generate_random_sparsity_v2]
for _generator in cfg_generator: for _generator in cfg_generator:
net = Model(pretrained=True, progress=False) net = Model(pretrained=False, progress=False)
cfg_list = _generator(net) cfg_list = _generator(net)
print('\n\nModel:', model_name) print('\n\nModel:', model_name)
......
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