A Computer-implemented Method of Training a Graph Neural Network