зеркало из https://github.com/microsoft/nni.git
Родитель
4959051758
Коммит
f9440cce54
|
@ -72,7 +72,6 @@ if __name__ == '__main__':
|
|||
|
||||
config_list = [{
|
||||
'op_types': ['Conv2d'],
|
||||
'op_names_re': ['features.*'],
|
||||
'lambda': 0.1,
|
||||
'apply_method': 'mse',
|
||||
}]
|
||||
|
|
|
@ -94,7 +94,6 @@ if __name__ == '__main__':
|
|||
|
||||
d_config_list = [{
|
||||
'op_types': ['Conv2d'],
|
||||
'op_names_re': ['features.*'],
|
||||
'lambda': 0.1,
|
||||
'apply_method': 'mse',
|
||||
}]
|
||||
|
|
|
@ -35,7 +35,6 @@ if __name__ == '__main__':
|
|||
|
||||
config_list = [{
|
||||
'op_types': ['Conv2d'],
|
||||
'op_names_re': ['features.*'],
|
||||
'sparse_ratio': 0.5
|
||||
}]
|
||||
dummy_input = torch.rand(8, 3, 224, 224).to(device)
|
||||
|
|
|
@ -32,7 +32,6 @@ if __name__ == '__main__':
|
|||
|
||||
config_list = [{
|
||||
'op_types': ['Conv2d'],
|
||||
'op_names_re': ['features.*'],
|
||||
'sparse_ratio': 0.5
|
||||
}]
|
||||
dummy_input = torch.rand(8, 3, 224, 224).to(device)
|
||||
|
|
|
@ -30,7 +30,6 @@ if __name__ == '__main__':
|
|||
|
||||
config_list = [{
|
||||
'op_types': ['Conv2d'],
|
||||
'op_names_re': ['features.*'],
|
||||
'sparse_ratio': 0.5
|
||||
}]
|
||||
dummy_input = torch.rand(8, 3, 224, 224).to(device)
|
||||
|
|
|
@ -30,7 +30,6 @@ if __name__ == '__main__':
|
|||
|
||||
config_list = [{
|
||||
'op_types': ['Conv2d'],
|
||||
'op_names_re': ['features.*'],
|
||||
'sparse_ratio': 0.5
|
||||
}]
|
||||
dummy_input = torch.rand(8, 3, 224, 224).to(device)
|
||||
|
|
Загрузка…
Ссылка в новой задаче