[GraphBolt] Labor example - Githubissues

dmlc / dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

http://dgl.ai

Apache License 2.0

13.41k stars 3k forks source link

[GraphBolt] Labor example #7437

Closed mfbalin closed 3 months ago

mfbalin commented 4 months ago

Description

This example is meant to showcase the GPU feature cache and its interaction with the labor parameters layer_dependency and batch_dependency. Since this will replace the official labor example, performance is of utmost importance. I also use early stopping to show that you can get good accuracies with labor sampler and a custom GraphSAGE model.

Checklist

Please feel free to remove inapplicable items for your PR.

[ ] The PR title starts with [$CATEGORY] (such as [NN], [Model], [Doc], [Feature]])
[ ] I've leverage the tools to beautify the python and c++ code.
[ ] The PR is complete and small, read the Google eng practice (CL equals to PR) to understand more about small PR. In DGL, we consider PRs with less than 200 lines of core code change are small (example, test and documentation could be exempted).
[ ] All changes have test coverage
[ ] Code is well-documented
[ ] To the best of my knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change
[ ] Related issue is referred in this PR
[ ] If the PR is for a new model/paper, I've updated the example index here.

Changes

dgl-bot commented 4 months ago

To trigger regression tests:

@dgl-bot run [instance-type] [which tests] [compare-with-branch]; For example: @dgl-bot run g4dn.4xlarge all dmlc/master or @dgl-bot run c5.9xlarge kernel,api dmlc/master

dgl-bot commented 4 months ago

Commit ID: 4572aa596d3e9cfdda2d6966a1b288f59705ea51

Build ID: 1

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: abc99dd6a6d364f8ae06e2a0efa0e1a02bda4750

Build ID: 2

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: 6e5fe74bc826d2136855e61feeb6fad22509fe5f

Build ID: 3

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: c2397cb57465a565092d29554029d2ab2b64233e

Build ID: 4

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: 05b1f2663a4382e51c983f10c2f95020ff317c76

Build ID: 5

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: b8e5be015f24371ac9283c77896956a0b46fb4af

Build ID: 6

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: ae3ef09ed960cfe3240efcd1f7b189c58a0c2766

Build ID: 7

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: 0614df4a303bc3aad097b3291e039a4e8f23049d

Build ID: 8

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

mfbalin commented 3 months ago

@frozenbugs I have moved the example under the PyG folder. Can you provide review for this PR? If DGL functionality is added later, it can be moved outside of the PyG examples folder.

dgl-bot commented 3 months ago

Commit ID: 60c400382776f38eb44ba9bbc5800fe9e5622afa

Build ID: 9

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 3 months ago

Commit ID: c8b94ed218fd2e7c608108e8b0fac28853752415

Build ID: 10

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link