File size: 430 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
model: mattergen
corruption: csp
loss_fn:
_target_: model.common.loss.MaterialsLoss
reduce: sum
include_pos: True
include_cell: True
include_atomic_numbers: False
weights:
cell: 1.0
pos: 0.1
pre_corruption_fn:
_target_: onescience.modules.embedding.mattergen_property_embeddings.SetEmbeddingType
p_unconditional: 0.2
dropout_fields_iid: false |