Skip to content

Request for MP-20-only pretrained CDVAE checkpoint #72

Description

@wtz01

Hi, thank you for making CDVAE publicly available.

I have been studying the repository and the MP-20 setup. From my understanding, the repository provides the MP-20 dataset configuration and the evaluation pipeline for a trained CDVAE model, while the files under cdvae/prop_models/mp20 correspond to the supervised property-prediction model used for evaluation, rather than the main generative CDVAE checkpoint.

I would like to use CDVAE as a pretrained crystal encoder-decoder for Materials Project structures. For this reason, I am specifically looking for the main CDVAE checkpoint trained only on the MP-20 dataset, i.e. the checkpoint that would be used as MODEL_PATH in scripts/evaluate.py for reconstruction / generation.

May I ask whether such an MP-20-only pretrained checkpoint is available? If so, could you please share it or point me to the corresponding download location?

Thank you very much for your help.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions