"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "321f9791d6a491ed140fd2cd26f56f45bbaa9f4a"
Unverified Commit 38077e29 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

installation and doc fix (#2379)

parent edb97877
...@@ -36,6 +36,7 @@ After the ``conda`` environment is activated, run one of the following commands. ...@@ -36,6 +36,7 @@ After the ``conda`` environment is activated, run one of the following commands.
conda install -c dglteam dgl-cuda10.0 # For CUDA 10.0 Build conda install -c dglteam dgl-cuda10.0 # For CUDA 10.0 Build
conda install -c dglteam dgl-cuda10.1 # For CUDA 10.1 Build conda install -c dglteam dgl-cuda10.1 # For CUDA 10.1 Build
conda install -c dglteam dgl-cuda10.2 # For CUDA 10.2 Build conda install -c dglteam dgl-cuda10.2 # For CUDA 10.2 Build
conda install -c dglteam dgl-cuda11.0 # For CUDA 11.0 Build
Install from pip Install from pip
...@@ -56,6 +57,7 @@ For CUDA builds, run one of the following commands and specify the CUDA version. ...@@ -56,6 +57,7 @@ For CUDA builds, run one of the following commands and specify the CUDA version.
pip install dgl-cu100 # For CUDA 10.0 Build pip install dgl-cu100 # For CUDA 10.0 Build
pip install dgl-cu101 # For CUDA 10.1 Build pip install dgl-cu101 # For CUDA 10.1 Build
pip install dgl-cu102 # For CUDA 10.2 Build pip install dgl-cu102 # For CUDA 10.2 Build
pip install dgl-cu110 # For CUDA 11.0 Build
For the most current nightly build from master branch, run one of the following commands. For the most current nightly build from master branch, run one of the following commands.
...@@ -67,6 +69,7 @@ For the most current nightly build from master branch, run one of the following ...@@ -67,6 +69,7 @@ For the most current nightly build from master branch, run one of the following
pip install --pre dgl-cu100 # For CUDA 10.0 Build pip install --pre dgl-cu100 # For CUDA 10.0 Build
pip install --pre dgl-cu101 # For CUDA 10.1 Build pip install --pre dgl-cu101 # For CUDA 10.1 Build
pip install --pre dgl-cu102 # For CUDA 10.2 Build pip install --pre dgl-cu102 # For CUDA 10.2 Build
pip install --pre dgl-cu110 # For CUDA 11.0 Build
.. _install-from-source: .. _install-from-source:
......
...@@ -236,6 +236,23 @@ class EdgeDataLoader: ...@@ -236,6 +236,23 @@ class EdgeDataLoader:
of blocks as computation dependency of the said minibatch for edge classification, of blocks as computation dependency of the said minibatch for edge classification,
edge regression, and link prediction. edge regression, and link prediction.
For each iteration, the object will yield
* A tensor of input nodes necessary for computing the representation on edges, or
a dictionary of node type names and such tensors.
* A subgraph that contains only the edges in the minibatch and their incident nodes.
Note that the graph has an identical metagraph with the original graph.
* If a negative sampler is given, another graph that contains the "negative edges",
connecting the source and destination nodes yielded from the given negative sampler.
* A list of blocks necessary for computing the representation of the incident nodes
of the edges in the minibatch.
For more details, please refer to :ref:`guide-minibatch-edge-classification-sampler`
and :ref:`guide-minibatch-link-classification-sampler`.
Parameters Parameters
---------- ----------
g : DGLGraph g : DGLGraph
...@@ -301,8 +318,9 @@ class EdgeDataLoader: ...@@ -301,8 +318,9 @@ class EdgeDataLoader:
>>> reverse_eids = torch.cat([torch.arange(E, 2 * E), torch.arange(0, E)]) >>> reverse_eids = torch.cat([torch.arange(E, 2 * E), torch.arange(0, E)])
Note that the sampled edges as well as their reverse edges are removed from Note that the sampled edges as well as their reverse edges are removed from
computation dependencies of the incident nodes. This is a common trick to avoid computation dependencies of the incident nodes. That is, the edge will not
information leakage. involve in neighbor sampling and message aggregation. This is a common trick
to avoid information leakage.
>>> sampler = dgl.dataloading.MultiLayerNeighborSampler([15, 10, 5]) >>> sampler = dgl.dataloading.MultiLayerNeighborSampler([15, 10, 5])
>>> dataloader = dgl.dataloading.EdgeDataLoader( >>> dataloader = dgl.dataloading.EdgeDataLoader(
......
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