admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 12:09:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

change device in test

Browse Source
This commit is contained in:
Pepijn
2025-09-10 19:50:49 +02:00
parent bd5c264c49
commit 9ea8bd029c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_pi0_openpi.py
+1 -1
View File
@@ -13,7 +13,7 @@ def test_policy_instantiation():
print("Testing PI0OpenPI policy instantiation...") print("Testing PI0OpenPI policy instantiation...")
# Create config # Create config
config = PI0OpenPIConfig(action_dim=7, state_dim=14, device="cpu", dtype="float32") config = PI0OpenPIConfig(action_dim=7, state_dim=14, dtype="float32")
# Create dummy dataset stats # Create dummy dataset stats
dataset_stats = { dataset_stats = {