Hugging Face's logo

collins909
/
my-diffusion-model

Model card Files
xet
 Community
my-diffusion-model
File size: 759 Bytes
e00756d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
{
  "model_type": "conditional_ddpm_camels",
  "in_channels": 1,
  "out_channels": 1,
  "image_size": 256,
  "label_dim": 2,
  "label_names": [
    "Omega_m",
    "sigma_8"
  ],
  "base_channels": 64,
  "channel_multipliers": [
    1,
    2,
    4,
    8
  ],
  "attention_levels": [
    2,
    3
  ],
  "dropout": 0.1,
  "timesteps": 1500,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "schedule_type": "linear",
  "ddim_steps_default": 50,
  "framework": "pytorch",
  "library_name": "pytorch",
  "training_meta": {
    "epochs": 100,
    "batch_size": 8,
    "lr": 0.0002,
    "ema_decay": 0.9999,
    "seed": 42
  },
  "label_mu": [
    0.3308129608631134,
    0.7831979990005493
  ],
  "label_std": [
    0.1140434592962265,
    0.12279357761144638
  ]
}