There are models that do not work.

[jihyeon-song]

While experimenting with various models while adjusting SbmBeamHandler.benchmarker_wrappers in sbm_config_mwe.gin, I found that models such as GIN, MLP, APPNP, SGC, and GATv2 in basic_gnn.py did not run except for GAT and GCN GraphSAGE. All raises "TypeError: 'NoneType' object is not subscriptable [while running 'Benchmark Simple GCN.']".
How to solve this case?
I modified SbmBeamHandler.benchmarker_wrappers as below to run ./main_local_mwe.sh
SbmBeamHandler.benchmarker_wrappers = [
@GAT_/NNNodeBenchmark,
@GIN_/NNNodeBenchmark,
@APPNP_/NNNodeBenchmark,
@GATv2_/NNNodeBenchmark,
]

[jpalowitch]

Hi, sorry for the very long delay. To add additional models, you also have to modify sbm_model_config_tuning_mwe.gin. Copy the blocks for GCN and GraphSAGE, and adjust the keys/values as needed.

Note that some models (such as APPNP) have additional parameters that you can set. See the other model_config_tuning files for examples.