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

fix test

Browse Source
This commit is contained in:
Pepijn
2025-10-13 17:59:59 +02:00
parent 4c40be57d8
commit 43bef1d91c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/training/test_multi_gpu.py
+2 -2
View File
@@ -119,7 +119,7 @@ class TestMultiGPUTraining:
config_args = [
"--dataset.repo_id=lerobot/pusht",
"--dataset.episodes=[0]",
"--policy=act",
"--policy.type=act",
"--policy.device=cuda",
f"--output_dir={output_dir}",
"--batch_size=4",
@@ -157,7 +157,7 @@ class TestMultiGPUTraining:
config_args = [
"--dataset.repo_id=lerobot/pusht",
"--dataset.episodes=[0]",
"--policy=act",
"--policy.type=act",
"--policy.device=cuda",
f"--output_dir={output_dir}",
"--batch_size=4",