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

fixup! fixup! fixup! fixup! fixup! Turn off compilation for pi0/pi05

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-07 22:59:29 +07:00
parent 3c484a77f6
commit a230e7424d
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rtc/eval_dataset.py
+6 -1
View File
@@ -482,7 +482,12 @@ class RTCEvaluator:
logging.info(f" Tracked {len(rtc_tracked_steps)} steps with RTC") logging.info(f" Tracked {len(rtc_tracked_steps)} steps with RTC")
# Save num_steps before destroying policy (needed for plotting) # Save num_steps before destroying policy (needed for plotting)
num_steps = policy_rtc_policy.config.num_steps try:
num_steps = policy_rtc_policy.config.num_steps
except Exception as e:
logging.error(f" Error getting num_steps: {e}")
num_steps = policy_rtc_policy.config.num_inference_steps
logging.warning(f" Using num_inference_steps: {num_steps} instead of num_steps")
# Destroy policy_rtc after final use # Destroy policy_rtc after final use
self._destroy_policy(policy_rtc_policy, "policy_rtc") self._destroy_policy(policy_rtc_policy, "policy_rtc")