admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-27 06:29:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(config): update vision encoder model name to lerobot/resnet10

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-18 15:15:59 +02:00
parent d7e25c8326
commit e4c1a8472d
4 changed files with 3 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_sac_policy.py
+1 -1
View File
@@ -304,7 +304,7 @@ def test_sac_training_with_visual_input(batch_size: int, state_dim: int, action_
@pytest.mark.parametrize(
"batch_size,state_dim,action_dim,vision_encoder_name",
[(1, 6, 6, "helper2424/resnet10"), (1, 6, 6, "facebook/convnext-base-224")],
[(1, 6, 6, "lerobot/resnet10"), (1, 6, 6, "facebook/convnext-base-224")],
)
@pytest.mark.skipif(not TRANSFORMERS_AVAILABLE, reason="Transformers are not installed")
def test_sac_policy_with_pretrained_encoder(