Skip to content

Commit 7b189dd

Browse files
committed
refactor(learner): Remove normalization from cached image features retrieval
- Simplified the retrieval of observation features by removing the normalization step from the `get_cached_image_features` method calls. - This change enhances clarity and aligns with the recent updates to policy processors.
1 parent d198ec2 commit 7b189dd

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/lerobot/scripts/rl/learner.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1049,10 +1049,8 @@ def get_observation_features(
10491049
return None, None
10501050

10511051
with torch.no_grad():
1052-
observation_features = policy.actor.encoder.get_cached_image_features(observations, normalize=True)
1053-
next_observation_features = policy.actor.encoder.get_cached_image_features(
1054-
next_observations, normalize=True
1055-
)
1052+
observation_features = policy.actor.encoder.get_cached_image_features(observations)
1053+
next_observation_features = policy.actor.encoder.get_cached_image_features(next_observations)
10561054

10571055
return observation_features, next_observation_features
10581056

0 commit comments

Comments
 (0)