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