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

dont push to hub in multi gpu tests

Browse Source
This commit is contained in:
Pepijn
2025-10-14 10:06:32 +02:00
parent 43bef1d91c
commit 252bca9354
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/training/test_multi_gpu.py
+2
View File
@@ -121,6 +121,7 @@ class TestMultiGPUTraining:
"--dataset.episodes=[0]", "--dataset.episodes=[0]",
"--policy.type=act", "--policy.type=act",
"--policy.device=cuda", "--policy.device=cuda",
"--policy.push_to_hub=false",
f"--output_dir={output_dir}", f"--output_dir={output_dir}",
"--batch_size=4", "--batch_size=4",
"--steps=10", "--steps=10",
@@ -159,6 +160,7 @@ class TestMultiGPUTraining:
"--dataset.episodes=[0]", "--dataset.episodes=[0]",
"--policy.type=act", "--policy.type=act",
"--policy.device=cuda", "--policy.device=cuda",
"--policy.push_to_hub=false",
f"--output_dir={output_dir}", f"--output_dir={output_dir}",
"--batch_size=4", "--batch_size=4",
"--steps=20", "--steps=20",