hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

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

Browse Source
This commit is contained in:
Eugene Mironov
2025-02-07 00:39:51 +07:00
committed by Michel Aractingi
parent 12c13e320e
commit af769abd8d
2 changed files with 88 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/configs/policy/hilserl_classifier.yaml
View File

@@ -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