Skip to content

GitLab

Switch branch/tag
  1. 25 Aug, 2022 1 commit
    • Chang Liu's avatar
      [Example][Refactor] Refactor RGCN example (#4327) · 40a2f3c7
      Chang Liu authored 
      * Refactor full graph entity classification

* Refactor rgcn with sampling

* README update

* Update

* Results update

* Respect default setting of self_loop=false in entity.py

* Update

* Update README

* Update for multi-gpu

* Update
      40a2f3c7
  3. 13 Jul, 2022 1 commit
  5. 11 May, 2022 1 commit
  7. 23 Feb, 2022 1 commit
    • Minjie Wang's avatar
      [NN] Rework RelGraphConv and HGTConv (#3742) · 0227ddfb
      Minjie Wang authored 
      * WIP: TypedLinear and new RelGraphConv

* wip

* further simplify RGCN

* a bunch of tweak for performance; add basic cpu support

* update on segmm

* wip: segment.cu

* new backward kernel works

* fix a bunch of bugs in kernel; leave idx_a for future

* add nn test for typed_linear

* rgcn nn test

* bugfix in corner case; update RGCN README

* doc

* fix cpp lint

* fix lint

* fix ut

* wip: hgtconv; presorted flag for rgcn

* hgt code and ut; WIP: some fix on reorder graph

* better typed linear init

* fix ut

* fix lint; add docstring
      0227ddfb
  9. 17 Feb, 2022 1 commit