File size: 482 Bytes
f15d29e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | _target_: model.diffusion.diffusion_module.DiffusionModule
loss_fn:
_target_: model.common.loss.MaterialsLoss
reduce: sum
include_pos: True
include_cell: True
include_atomic_numbers: True
d3pm_hybrid_lambda: 0.01
weights:
cell: 1.0
pos: 0.1
atomic_numbers: 1.0
model: mattergen
corruption: default
pre_corruption_fn:
_target_: onescience.modules.embedding.mattergen_property_embeddings.SetEmbeddingType
p_unconditional: 0.2
dropout_fields_iid: false
|