fix model export. Update tf2onnx to 1.9.2

This commit is contained in:
iperov 2021-11-20 16:05:44 +04:00
parent 4b6bf003b8
commit 388964e8d0
4 changed files with 4 additions and 4 deletions

View file

@ -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