diff --git a/ppdet/engine/export_utils.py b/ppdet/engine/export_utils.py index 01184136b9..bd0400e177 100644 --- a/ppdet/engine/export_utils.py +++ b/ppdet/engine/export_utils.py @@ -424,7 +424,10 @@ def get_dynamic_shapes(hpi_shape): "BlazeFace", "BlazeFace-FPN-SSH", "PP-YOLOE_seg", - "SOLOv2" + "SOLOv2", + "PP-DocLayout-L", + "PP-DocLayout_plus-L", + "PP-DocBlockLayout", ] if any(name in pdx_model_name for name in model_names_required_imgsize): shapes["im_shape"] = [[batch_sizes[0], 2], [batch_sizes[1], 2], [batch_sizes[2], 2]]