Skip to content

Latest commit

 

History

History
20 lines (19 loc) · 710 Bytes

File metadata and controls

20 lines (19 loc) · 710 Bytes

ogbn-arxiv

Default

--use_gpu False 
--self_loop False
--block res+ 	#options: [plain, res, res+]
--conv gen
--gcn_aggr max 	#options: [max, mean, add, softmax, softmax_sg, softmax_sum, power, power_sum]
--num_layers 3
--mlp_layers 1
--norm batch
--hidden_channels 128
--epochs 500
--lr 0.001
--dropout 0.5

ResGEN

Train

python main.py --use_gpu --self_loop --num_layers 28 --block res+ --gcn_aggr softmax_sg --t 0.1

Test (use pre-trained model, download from Google Drive)

python test.py --use_gpu --self_loop --num_layers 28 --block res+ --gcn_aggr softmax_sg --t 0.1