您当前的位置: 首页 > 

静静喜欢大白

暂无认证

  • 2浏览

    0关注

    521博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

dgl-数据集修改默认下载地址

静静喜欢大白 发布时间:2021-08-07 15:14:49 ,浏览量:2

1、简介

在运行dgl相关代码的时候数据总是默认下载到/home/user/.dgl/下面,但是我想将数据集下载到我想下载的位置的话,该怎么解决呢?

2、解决

只需要修改_init_这个函数,修改raw_dir这个默认路径即可

代码如下

def __init__(self):
        super().__init__(name='ACM', url=_get_dgl_url('dataset/ACM.mat'))
修改成

def __init__(self, raw_dir = './dataset'): #./表示上一级路径
        super().__init__(name='ACM', url=_get_dgl_url('dataset/ACM.mat'), raw_dir=raw_dir)

关注
打赏
1510642601
查看更多评论
立即登录/注册

微信扫码登录

0.2408s