[hotfix] pytest error of act_act example (#12093)
* [hotfix] pytest error of act_act example * remove needless import
This commit is contained in:
parent
bbf54e01e7
commit
b371e74735
4
tools/pot/tests/data/reference_models/act_act_example_DefaultQuantization_pytorch.xml
Executable file → Normal file
4
tools/pot/tests/data/reference_models/act_act_example_DefaultQuantization_pytorch.xml
Executable file → Normal file
@ -1,3 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:e6eb24c47d38ae992eef207a5c4e873db496a63eed42ac5469df4f2afc756f48
|
||||
size 16835
|
||||
oid sha256:f44cbc66f71d53879ead60e9db65a80c3d9b572323aca87265567a01ca77426d
|
||||
size 16673
|
||||
|
@ -15,14 +15,14 @@ from .utils.check_graph import check_model
|
||||
from .utils.config import merge_configs
|
||||
|
||||
TEST_MODELS = [
|
||||
('resnet_example', 'pytorch')
|
||||
('resnet_example', 'pytorch', 'Ranger', 'ANY'),
|
||||
('resnet_example', 'pytorch', 'DefaultQuantization', 'CPU'),
|
||||
('resnet_example', 'pytorch', 'RangeSupervision', 'ANY'),
|
||||
('act_act_example', 'pytorch', 'DefaultQuantization', 'GNA')
|
||||
]
|
||||
|
||||
|
||||
@pytest.fixture(scope='module', params=TEST_MODELS,
|
||||
ids=['{}_{}_{}'.format(*m) for m in TEST_MODELS])
|
||||
ids=['_'.join(m) for m in TEST_MODELS])
|
||||
def _params(request):
|
||||
return request.param
|
||||
|
||||
@ -32,7 +32,7 @@ def test_graph_with_stats(_params, tmp_path, models):
|
||||
|
||||
algorithm_config = Dict({
|
||||
'algorithms': [{
|
||||
'name': 'algo_name',
|
||||
'name': algo_name,
|
||||
'params': {
|
||||
'target_device': target_device,
|
||||
'preset': 'performance',
|
||||
|
Loading…
Reference in New Issue
Block a user