refine

ci/test.sh

@@ -12,5 +12,34 @@ python3 setup.py install
 # python3 examples/tensorrt_qat/test_lenet_qat_train.py
 # python3 -m pytest -s examples/tensorrt_qat/test_lenet_qat.py
 python3 -m pytest examples/oneflow2onnx
+python3 -m pytest examples/x2oneflow/pytorch2oneflow/nodes
+python3 -m pytest examples/x2oneflow/pytorch2oneflow/models
+python3 -m pytest examples/x2oneflow/tensorflow2oneflow/nodes
+python3 -m pytest examples/x2oneflow/tensorflow2oneflow/models
+python3 -m pytest examples/x2oneflow/paddle2oneflow/nodes
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_alexnet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_darknet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_densenet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_dpn.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_efficientnet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_ghostnet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_googlenet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_inceptionv3.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_inceptionv4.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_mobilenetv1.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_mobilenetv2.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_mobilenetv3.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_regnet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_repvgg.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_res2net.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_resnet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_resnext.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_se_resnext.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_shufflenet_v2.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_squeezenet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_vggnet.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_vision_transformer.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_xception_deeplab.py
+python3 -m pytest examples/x2oneflow/paddle2oneflow/models/test_xception.py
 ./test_code_gen.sh
 

oneflow_onnx/x2oneflow/util.py

@@ -146,6 +146,8 @@ def load_pytorch_module_and_check(
 
     if oneflow_code_gen_flag == True and len(input_size) == 4:
         oneflow_code_gen_func(input_size, model_weight_save_dir)
+        flow.clear_default_session()
+        return
         
     if train_flag == False:
         pt_module.eval()
@@ -236,6 +238,8 @@ def load_paddle_module_and_check(
 
     if oneflow_code_gen_flag == True and len(input_size) == 4:
         oneflow_code_gen_func(input_size, model_weight_save_dir)
+        flow.clear_default_session()
+        return
     
     if train_flag == False:
         pd_module.eval()
@@ -331,6 +335,8 @@ def load_tensorflow2_module_and_check(
 
     if oneflow_code_gen_flag == True and len(input_size) == 4:
         oneflow_code_gen_func(input_size, model_weight_save_dir)
+        flow.clear_default_session()
+        return
     
     if train_flag == True:
         flow_res = job_train(ipt1)