在d4rl代码库中,默认数据集下载的位置在offline_env.py中定义:
def set_dataset_path(path):
global DATASET_PATH
DATASET_PATH = path
os.makedirs(path, exist_ok=True)
set_dataset_path(os.environ.get('D4RL_DATASET_DIR', os.path.expanduser('~/.d4rl/datasets')))
即在环境变量D4RL_DATASET_DIR没有定义的情况下,默认为~/.d4rl/datasets
将数据集迁移至指定地点,需要设置系统环境变量
sudo vim /etc/profile
export D4RL_DATASET_DIR=$your path$
source /etc/profile
- 将数据集迁移过去
mv ~/.d4rl/datasets $your path$