Graph_neural_network loading ...