Skip to content

Commit 59e5e45

Browse files
chyomin06fracape
authored andcommitted
[fix] returned feature tensor in base pipeline
1 parent 9a84c5f commit 59e5e45

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

compressai_vision/pipelines/base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,10 +239,11 @@ def _post_process_loaded_features(features, n_bits, datacatalog_name):
239239
for key, data in features["data"].items():
240240
out = min_max_inv_normalization(data, minv, maxv, bitdepth=n_bits)
241241
data_features[key] = out.to(torch.float32)
242+
243+
features["data"] = data_features
242244
else:
243245
raise NotImplementedError
244246

245-
features["data"] = data_features
246247
return features
247248

248249
def _from_input_to_features(

0 commit comments

Comments
 (0)