Skip to content

Commit

Permalink
[fix] Fixed the issue in the runtime result sorting order of OIV6
Browse files Browse the repository at this point in the history
  • Loading branch information
sarhakor authored and fracape committed Nov 26, 2024
1 parent 8788756 commit 55ea231
Showing 1 changed file with 13 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -222,6 +222,19 @@ def __call__(
out_res["input_size"] = dec_features["input_size"][0]
output_list.append(out_res)

if not self.configs["codec"]["decode_only"]:
accum_enc_by_module = {
key: accum_enc_by_module[key]
for key in ["feature_reduction", "conversion", "inner_codec"]
if key in accum_enc_by_module
}
if not self.configs["codec"]["encode_only"]:
accum_dec_by_module = {
key: accum_dec_by_module[key]
for key in ["inner_codec", "conversion", "feature_restoration"]
if key in accum_dec_by_module
}

# if dec_only is True, accum_enc_by_module is None
self.add_time_details("encode", accum_enc_by_module)
# if enc_only is True, accum_dec_by_module is None
Expand Down

0 comments on commit 55ea231

Please sign in to comment.