free-energy-gnn Requirements You may need to install following packages for the network to train: pytorch pytorch-geometric (version 1.5.0) I should provide the requirements.txt file soon (or the Docker, even better)