IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix problems with pytorch models passed to convert_model (#17255)

Browse Source 
* Do eval() only for torch Module

* Add test

* Support decoder in convert_model

* Enable tests
This commit is contained in:
Maxim Vafin
2023-04-27 18:33:46 +04:00
committed by GitHub
parent 2b8a6ba99a
commit 1d443c6da6
4 changed files with 25 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/layer_tests/mo_python_api_tests/test_mo_convert_pytorch.py
+2 -3
View File
@@ -780,9 +780,8 @@ class TestMoConvertPyTorch(CommonMOConvertTest):
create_pytorch_nn_module_sample_list_of_tensors,
create_pytorch_jit_script_module,
create_pytorch_jit_script_function,
# Disabled due to Ticket-109430
#create_pytorch_nn_module_layout_list,
#create_pytorch_nn_module_layout_list_case2,
create_pytorch_nn_module_layout_list,
create_pytorch_nn_module_layout_list_case2,
create_pytorch_nn_module_mean_list,
create_pytorch_nn_module_mean_list_default_no_compression,
create_pytorch_nn_module_mean_list_compressin_enabled,