train_GNN_buckling.py