Skip to content

Commit 55ea231

Browse files
sarhakorfracape
authored andcommitted
[fix] Fixed the issue in the runtime result sorting order of OIV6
1 parent 8788756 commit 55ea231

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

compressai_vision/pipelines/split_inference/image_split_inference.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,19 @@ def __call__(
222222
out_res["input_size"] = dec_features["input_size"][0]
223223
output_list.append(out_res)
224224

225+
if not self.configs["codec"]["decode_only"]:
226+
accum_enc_by_module = {
227+
key: accum_enc_by_module[key]
228+
for key in ["feature_reduction", "conversion", "inner_codec"]
229+
if key in accum_enc_by_module
230+
}
231+
if not self.configs["codec"]["encode_only"]:
232+
accum_dec_by_module = {
233+
key: accum_dec_by_module[key]
234+
for key in ["inner_codec", "conversion", "feature_restoration"]
235+
if key in accum_dec_by_module
236+
}
237+
225238
# if dec_only is True, accum_enc_by_module is None
226239
self.add_time_details("encode", accum_enc_by_module)
227240
# if enc_only is True, accum_dec_by_module is None

0 commit comments

Comments
 (0)