admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-25 10:46:01 +00:00
Code Issues Packages Projects Releases Wiki 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 GitHub
parent a7db3959f5
commit a1d16fb400
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lerobot/common/policies/hilserl/classifier/configuration_classifier.py
+1 -1
View File
@@ -10,7 +10,7 @@ class ClassifierConfig:
num_classes: int = 2
hidden_dim: int = 256
dropout_rate: float = 0.1
model_name: str = "microsoft/resnet-50"
model_name: str = "helper2424/resnet10"
device: str = "cpu"
model_type: str = "cnn" # "transformer" or "cnn"
num_cameras: int = 2