Skip to content

GitLab

Commit cccc2339 authored by Haibin Lin's avatar Haibin Lin Committed by Minjie Wang
Browse files

[Dataset] update s3 links (#209)

parent e02acc4c
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
python/dgl/data/citation_graph.py
View file @ cccc2339
...@@ -15,9 +15,9 @@ import os, sys ...@@ -15,9 +15,9 @@ import os, sys
from .utils import download, extract_archive, get_download_dir from .utils import download, extract_archive, get_download_dir
_urls = { _urls = {
'cora' : 'https://s3.amazonaws.com/dgl-data/dataset/cora.zip', 'cora' : 'https://s3.us-east-2.amazonaws.com/dgl.ai/dataset/cora.zip',
'citeseer' : 'https://s3.amazonaws.com/dgl-data/dataset/citeseer.zip', 'citeseer' : 'https://s3.us-east-2.amazonaws.com/dgl.ai/dataset/citeseer.zip',
'pubmed' : 'https://s3.amazonaws.com/dgl-data/dataset/pubmed.zip', 'pubmed' : 'https://s3.us-east-2.amazonaws.com/dgl.ai/dataset/pubmed.zip',
} }
class CitationGraphDataset(object): class CitationGraphDataset(object):
......
python/dgl/data/tree.py
View file @ cccc2339
...@@ -17,7 +17,7 @@ import dgl.backend as F ...@@ -17,7 +17,7 @@ import dgl.backend as F
from dgl.data.utils import download, extract_archive, get_download_dir from dgl.data.utils import download, extract_archive, get_download_dir
_urls = { _urls = {
'sst' : 'https://s3.amazonaws.com/dgl-data/dataset/sst.zip', 'sst' : 'https://s3.us-east-2.amazonaws.com/dgl.ai/dataset/sst.zip',
} }
SSTBatch = namedtuple('SSTBatch', ['graph', 'mask', 'wordid', 'label']) SSTBatch = namedtuple('SSTBatch', ['graph', 'mask', 'wordid', 'label'])
......
tutorials/models/5_dgmg.py
View file @ cccc2339
...@@ -713,7 +713,7 @@ class DGMG(DGMGSkeleton): ...@@ -713,7 +713,7 @@ class DGMG(DGMGSkeleton):
import torch.utils.model_zoo as model_zoo import torch.utils.model_zoo as model_zoo
# Download a pre-trained model state dict for generating cycles with 10-20 nodes. # Download a pre-trained model state dict for generating cycles with 10-20 nodes.
state_dict = model_zoo.load_url('https://s3.amazonaws.com/dgl-data/model/dgmg_cycles-5a0c40be.pth') state_dict = model_zoo.load_url('https://s3.us-east-2.amazonaws.com/dgl.ai/model/dgmg_cycles-5a0c40be.pth')
model = DGMG(v_max=20, node_hidden_size=16, num_prop_rounds=2) model = DGMG(v_max=20, node_hidden_size=16, num_prop_rounds=2)
model.load_state_dict(state_dict) model.load_state_dict(state_dict)
model.eval() model.eval()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment