atom-predict/egnn_v2/model/xj_new_2.ckpt