geodiff-drugs-dualenc / config.json
Nathan Lambert
Upload config.json
22f7630
raw
history blame contribute delete
401 Bytes
{
"_class_name": "DualEncoderEpsNetwork",
"_diffusers_version": "0.0.4",
"type": "diffusion",
"network": "dualenc",
"hidden_dim": 128,
"num_convs": 6,
"num_convs_local": 4,
"cutoff": 10.0,
"mlp_act": "relu",
"beta_schedule": "sigmoid",
"beta_start": 1e-7,
"beta_end": 2e-3,
"num_diffusion_timesteps": 5000,
"edge_order": 3,
"edge_encoder": "mlp",
"smooth_conv": true
}