cyx
2 years ago
1 changed files with 3 additions and 3 deletions
Unified View
Diff Options
-
+3 -3transformer/serve.yaml
+ 3
- 3
transformer/serve.yaml
View File
| @@ -132,7 +132,7 @@ single: | |||||
| outputs: | outputs: | ||||
| - name: dets | - name: dets | ||||
| type: fp32 | type: fp32 | ||||
| shape: [ -1, -1, 5] | |||||
| shape: [ -1, -1, -1] | |||||
| - name: labels | - name: labels | ||||
| type: int64 | type: int64 | ||||
| shape: [ -1, -1] | shape: [ -1, -1] | ||||
| @@ -141,10 +141,10 @@ single: | |||||
| shape: [ -1, 1280] | shape: [ -1, 1280] | ||||
| - name: entropy | - name: entropy | ||||
| type: fp32 | type: fp32 | ||||
| shape: [ -1, -1, 256] | |||||
| shape: [ -1, 1] | |||||
| - name: learning_loss | - name: learning_loss | ||||
| type: fp32 | type: fp32 | ||||
| shape: [ -1, -1] | |||||
| shape: [ -1, 1] | |||||
| #postprocess对应的backend固定为 apulis.postprocessor | #postprocess对应的backend固定为 apulis.postprocessor | ||||
| postprocess: | postprocess: | ||||
| type: python | type: python | ||||