atom-predict/egnn_v2/model/last.ckpt