[fix] datatype conversion on for image split inference

Author: chyomin06

compressai_vision/pipelines/split_inference/image_split_inference.py

@@ -119,8 +119,10 @@ def __call__(
                 self.update_time_elapsed("nn_part_1", (time_measure() - start))
 
                 # datatype conversion
-                # featureT["data"] = {k : v.type(getattr(torch, self.datatype)) for k, v in featureT["data"].items()}
-
+                featureT["data"] = {
+                    k: v.type(getattr(torch, self.datatype))
+                    for k, v in featureT["data"].items()
+                }
                 featureT["org_input_size"] = org_img_size
 
                 start = time_measure()