wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Auto set device

This commit is contained in:
Simon Alibert 2024-04-17 16:42:55 +02:00
parent 75d34548a5
commit a870907266
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/utils.py
View File

@ -1,6 +1,6 @@
import os import torch
# Pass this as the first argument to init_hydra_config. # Pass this as the first argument to init_hydra_config.
DEFAULT_CONFIG_PATH = "lerobot/configs/default.yaml" DEFAULT_CONFIG_PATH = "lerobot/configs/default.yaml"
DEVICE = os.environ.get('LEROBOT_TESTS_DEVICE', "cuda") DEVICE = "cuda" if torch.cuda.is_available() else "cpu"