Add DEVICE constant from LEROBOT_TESTS_DEVICE
This commit is contained in:
@@ -1,8 +1,10 @@
|
||||
import os
|
||||
import hydra
|
||||
from hydra import compose, initialize
|
||||
|
||||
CONFIG_PATH = "../lerobot/configs"
|
||||
|
||||
DEVICE = os.environ.get('LEROBOT_TESTS_DEVICE', "cuda")
|
||||
|
||||
def init_config(config_name="default", overrides=None):
|
||||
hydra.core.global_hydra.GlobalHydra.instance().clear()
|
||||
|
||||
Reference in New Issue
Block a user