Configuration Templates¶
Ready-to-use template configs live in the repository under docs/cfg_examples/.
Copy the relevant file, fill in your checkpoint and data paths, and run it. The
full packaged Hydra defaults these templates build on are under
src/MolecularDiffusion/configs/.
Training¶
Template |
Purpose |
|---|---|
|
Train an EDM diffusion model from scratch. |
|
Fine-tune from a pretrained diffusion checkpoint. |
|
Train a property regressor. |
|
Train a guidance (property) model. |
MolCraftDiff train docs/cfg_examples/train_diffusion
Generation¶
Template |
Mode ( |
|---|---|
|
|
|
|
|
|
|
|
|
|
|
|
MolCraftDiff generate docs/cfg_examples/gen_cfg
Every command accepts Hydra-style dotted overrides, e.g.
MolCraftDiff generate docs/cfg_examples/gen_cfg interference.num_generate=200.
See Model Architectures for choosing a training backbone and
Tutorials for end-to-end walkthroughs.