admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-24 13:09:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove leftover normalization calls from reward classifier predict_reward

Browse Source 
Fixes #2355
This commit is contained in:
Khalil Meftah
2026-04-13 13:29:00 +02:00
parent 11a0b0174f
commit 9422dc98c2
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/sac/reward_model/modeling_classifier.py
-4
View File
@@ -268,10 +268,6 @@ class Classifier(PreTrainedPolicy):
def predict_reward(self, batch, threshold=0.5): def predict_reward(self, batch, threshold=0.5):
"""Eval method. Returns predicted reward with the decision threshold as argument.""" """Eval method. Returns predicted reward with the decision threshold as argument."""
# Check for both OBS_IMAGE and OBS_IMAGES prefixes
batch = self.normalize_inputs(batch)
batch = self.normalize_targets(batch)
# Extract images from batch dict # Extract images from batch dict
images = [batch[key] for key in self.config.input_features if key.startswith(OBS_IMAGE)] images = [batch[key] for key in self.config.input_features if key.startswith(OBS_IMAGE)]