An attempt to recreate the results reported here.
Most of the code comes from the original repo with a few changes.
Recommended versions for torch and pytorch-lightning are respectively 1.12.1 and 1.6.4.
(The current notebook has only been trained for 30 epochs, to achieve results similar to the ones reported in the paper, 300 epochs are needed.)
NegarAskari/GraphBepi-Notebook
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|