fix model export. Update tf2onnx to 1.9.2

2021-11-20 16:05:44 +04:00
parent 4b6bf003b8
commit 388964e8d0
4 changed files with 4 additions and 4 deletions
@@ -625,7 +625,7 @@ class AMPModel(ModelBase):
                name='AMP',
                input_names=['in_face:0','morph_value:0'],
                output_names=['out_face_mask:0','out_celeb_face:0','out_celeb_face_mask:0'],
-                opset=9,
+                opset=12,
                output_path=output_path)

    #override
@@ -746,7 +746,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
                name='SAEHD',
                input_names=['in_face:0'],
                output_names=['out_face_mask:0','out_celeb_face:0','out_celeb_face_mask:0'],
-                opset=9,
+                opset=12,
                output_path=output_path)

    #override