admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 17:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

add small logging

Browse Source
This commit is contained in:
Pepijn
2025-11-25 22:54:35 +01:00
parent 599c2477c5
commit c66aef878c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/sarm/modeling_sarm.py
+1 -1
View File
@@ -309,7 +309,7 @@ class SARMRewardModel(PreTrainedPolicy):
self.config.num_stages = num_stages self.config.num_stages = num_stages
self.config.subtask_names = subtask_names self.config.subtask_names = subtask_names
logging.info(f"Auto-detected {num_stages} subtasks from dataset: {subtask_names}") logging.info(f"Auto-detected {num_stages} subtasks from dataset: {subtask_names}, using {num_stages} stages")
def to(self, device): def to(self, device):
"""Override to method to ensure all components move together.""" """Override to method to ensure all components move together."""