atom-predict/egnn_v2/model/xj.ckpt