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

[Port HIL-SERL] Add resnet-10 as default encoder for HIL-SERL (#696)

Browse Source 
Co-authored-by: Khalil Meftah <kmeftah.khalil@gmail.com>
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co>
Co-authored-by: Ke Wang <superwk1017@gmail.com>
This commit is contained in:
Eugene Mironov
2025-02-11 17:37:00 +07:00
committed by Michel Aractingi
parent f1af97dc9c
commit 3a07301365
3 changed files with 3 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

@@ -36,7 +36,7 @@ eval:
policy:
name: "hilserl/classifier/push_green_cube_hf_cropped_resized" #"hilserl/classifier/pick_place_lego_cube_1"
model_name: "facebook/convnext-base-224"
model_name: "helper2424/resnet10"
model_type: "cnn"
num_cameras: 2 # Has to be len(training.image_keys)