File tree Expand file tree Collapse file tree 1 file changed +13
-0
lines changed
compressai_vision/pipelines/split_inference Expand file tree Collapse file tree 1 file changed +13
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments