Spaces:
Sleeping
Sleeping
File size: 1,734 Bytes
96da58e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
"""
Set of global variables shared across robomimic
"""
# Sets debugging mode. Should be set at top-level script so that internal
# debugging functionalities are made active
DEBUG = False
# Whether to visualize the before & after of an observation randomizer
VISUALIZE_RANDOMIZER = False
# wandb entity (eg. username or team name)
WANDB_ENTITY = None
# wandb api key (obtain from https://wandb.ai/authorize)
# alternatively, set up wandb from terminal with `wandb login`
WANDB_API_KEY = None
### Slack Notifications ###
# Token for sending slack notifications
SLACK_TOKEN = None
# User ID for user that should receive slack notifications
SLACK_USER_ID = None
### Local Sync Settings ###
# By specifying this path, you can sync the most important results of training back to this folder
RESULTS_SYNC_PATH = None
# This will be automatically populated.
RESULTS_SYNC_PATH_ABS = None
### MagLev and NGC Cluster Settings ###
# Whether training is happening on MagLev / NGC (should set this on repos hosted in MagLev / NGC scratch space or in Docker)
USE_MAGLEV = False
USE_NGC = False
# When using MagLev / NGC, sync the most important results of training back to this directory in scratch space.
# This path should be relative to the base scratch space directory (for MagLev) or an absolute path (for NGC)
MAGLEV_SCRATCH_SYNC_PATH = None
NGC_SCRATCH_SYNC_PATH = None
try:
from robomimic.macros_private import *
except ImportError:
from robomimic.utils.log_utils import log_warning
import robomimic
log_warning(
"No private macro file found!"\
"\nIt is recommended to use a private macro file"\
"\nTo setup, run: python {}/scripts/setup_macros.py".format(robomimic.__path__[0])
)
|