[HIL-SERL port] Add Reward classifier benchmark tracking to chose best visual encoder (#688)

2025-02-07 00:39:51 +07:00
parent 729b4ed697
commit 434d1e0614
2 changed files with 88 additions and 3 deletions
--- a/lerobot/configs/policy/hilserl_classifier.yaml
+++ b/lerobot/configs/policy/hilserl_classifier.yaml
@@ -27,6 +27,8 @@ training:
  # image_keys: ["observation.images.top", "observation.images.wrist"]
  image_keys: ["observation.images.laptop", "observation.images.phone"]
  label_key: "next.reward"
+  profile_inference_time: false
+  profile_inference_time_iters: 20

 eval:
  batch_size: 16