tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix caching and dataset stats is optional

Browse Source
This commit is contained in:
AdilZouitine
2025-04-09 13:20:51 +00:00
committed by Michel Aractingi
parent 68c271ad25
commit e18274bc9a
3 changed files with 22 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/server/learner_server.py
View File

@@ -1026,8 +1026,8 @@ def get_observation_features(
return None, None
with torch.no_grad():
observation_features = policy.actor.encoder.get_image_features(observations)
next_observation_features = policy.actor.encoder.get_image_features(next_observations)
observation_features = policy.actor.encoder.get_image_features(observations, normalize=True)
next_observation_features = policy.actor.encoder.get_image_features(next_observations, normalize=True)
return observation_features, next_observation_features