Skip to content

why the model_type setting in training.yml is uni_o2? #22

@daisy-zzy

Description

@daisy-zzy

Thank you for your great work and open-source code!
According to the guide, to train the model from scratch, we have to run:
python scripts/train_diffusion.py configs/training.yml
However, the model_type setting in training.yml is uni_o2, which means the denoising process will use UniTransformerO2TwoUpdateGeneral instead of EGNN as the refine net. Is this a mistake or is it the common case in training? @guanjq

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