mirrors / dmlc--dgl

项目文件夹

文件

T

张天启 72ef642f29 [Example] Add sagpool example for pytorch backend (#2429 )

* add sagpool example for pytorch backend

* polish sagpool example for pytorch backend

* [Example] SAGPool: use std variance

* [Example] SAGPool: change to std

* add sagpool example to index page

* add graph property prediction tag to sagpool

Co-authored-by: zhangtianqi <tianqizh@amazon.com>

2020-12-28 17:44:55 +08:00

9 行

270 B

JSON

原始文件 永久链接 Blame 文件历史

 {
     "arch": ["hierarchical", "global"],
     "hidden": [16, 32, 64, 128],
     "pool_ratio": [0.25, 0.5],
     "lr": [1e-2, 5e-2, 1e-3, 5e-3, 1e-4, 5e-4],
     "weight_decay": [1e-2, 1e-3, 1e-4, 1e-5],
     "dataset": ["DD", "PROTEINS", "NCI1", "NCI109", "Mutagenicity"]
 }