File size: 402 Bytes
4ed7d0c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
{
  "_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": false
}