diff --git a/transformer/serve.yaml b/transformer/serve.yaml
index c8d4c28..1e4574d 100644
--- a/transformer/serve.yaml
+++ b/transformer/serve.yaml
@@ -132,7 +132,7 @@ single:
     outputs:
       - name: dets
         type: fp32
-        shape: [ -1, -1, 5]
+        shape: [ -1, -1, -1]
       - name: labels
         type: int64
         shape: [ -1, -1]
@@ -141,10 +141,10 @@ single:
         shape: [ -1, 1280]
       - name: entropy
         type: fp32
-        shape: [ -1, -1, 256]
+        shape: [ -1, 1]
       - name: learning_loss
         type: fp32
-        shape: [ -1, -1]
+        shape: [ -1, 1]
   #postprocess对应的backend固定为 apulis.postprocessor
   postprocess:
     type: python