lerobot/tests/utils.py

import torch

# Pass this as the first argument to init_hydra_config.
DEFAULT_CONFIG_PATH = "lerobot/configs/default.yaml"

DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
Auto set device 2024-04-17 22:42:55 +08:00			`import torch`
Add policies/factory, Add test, Add _self_ in config 2024-02-25 18:50:23 +08:00
revision 2024-03-28 02:33:48 +08:00			`# Pass this as the first argument to init_hydra_config.`
			`DEFAULT_CONFIG_PATH = "lerobot/configs/default.yaml"`
Add policies/factory, Add test, Add _self_ in config 2024-02-25 18:50:23 +08:00
Auto set device 2024-04-17 22:42:55 +08:00			`DEVICE = "cuda" if torch.cuda.is_available() else "cpu"`