Skip to content

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

@Timodric

Description

@Timodric

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.

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