d4rl修改默认数据集位置

在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
  1. 将数据集迁移过去
mv ~/.d4rl/datasets $your path$

猜你喜欢

转载自blog.csdn.net/CCCDeric/article/details/131032992