How to write a graph classification model using torch_geometric.graphgym #7393
-
|
How to write a graph classification model using torch_geometric.graphgym？There are too few references on the Internet |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 8 replies
-
|
I would like to have a code that compares multiple aggregations or pools |
Beta Was this translation helpful? Give feedback.
-
|
Did you take a look at https://github.com/pyg-team/pytorch_geometric/blob/master/graphgym/configs/pyg/example_graph.yaml? In the end, you just need to ensure that |
Beta Was this translation helpful? Give feedback.
-
|
Which version of PyTorch Lightning are you on? |
Beta Was this translation helpful? Give feedback.
-
|
pytorch-lightning 2.0.2 |
Beta Was this translation helpful? Give feedback.
-
|
I see. You may need to upgrade to PyG 2.3 for PyTorch Lightning 2.0 support. |
Beta Was this translation helpful? Give feedback.
-
|
thank，The script is now working properly！ |
Beta Was this translation helpful? Give feedback.
-
|
You could mimic the logic we have here to run it within your own Pythons script: https://github.com/pyg-team/pytorch_geometric/blob/master/graphgym/main.py |
Beta Was this translation helpful? Give feedback.
thank，The script is now working properly！
and one more question,
Does torch_geometric.graphgym only support starting tasks with scripts like this?
Not in some python compiler (such as PyCharm)?