File size: 830 Bytes
f15d29e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | _target_: model.diffusion.corruption.multi_corruption.MultiCorruption
sdes:
pos:
_target_: model.common.diffusion.corruption.NumAtomsVarianceAdjustedWrappedVESDE
wrapping_boundary: 1.0
sigma_max: 5.0
limit_info_key: num_atoms
cell:
_target_: model.common.diffusion.corruption.LatticeVPSDE.from_vpsde_config
vpsde_config:
beta_min: 0.1
beta_max: 20
limit_density: ${data_module.average_density}
limit_var_scaling_constant: 0.25
discrete_corruptions:
atomic_numbers:
_target_: model.diffusion.corruption.d3pm_corruption.D3PMCorruption
offset: 1
d3pm:
_target_: model.diffusion.d3pm.d3pm.MaskDiffusion
dim: 101
schedule:
_target_: model.diffusion.d3pm.d3pm.create_discrete_diffusion_schedule
kind: standard
num_steps: 1000
|