refactor(diffusion): remove normalization layer from input processing

This commit is contained in:
Adil Zouitine
2025-07-28 13:43:07 +02:00
committed by Steven Palma
parent 1feb7b5d88
commit f02ce69df0
@@ -124,7 +124,6 @@ class DiffusionPolicy(PreTrainedPolicy):
if ACTION in batch:
batch.pop(ACTION)
batch = self.normalize_inputs(batch)
if self.config.image_features:
batch = dict(batch) # shallow copy so that adding a key doesn't modify the original
batch[OBS_IMAGES] = torch.stack([batch[key] for key in self.config.image_features], dim=-4)