d4rl修改默认数据集位置

5 篇文章 0 订阅
4 篇文章 0 订阅
在d4rl代码库中,当D4RL_DATASET_DIR环境变量未设置时,数据集默认存储在~/.d4rl/datasets。要改变这个位置,需编辑系统环境变量,在/etc/profile中添加出口令并更新,然后将现有数据集移动到新路径。
摘要由CSDN通过智能技术生成

在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$
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值