Hugging Face's logo

jz666
/
dit_s2

Model card Files
xet
 Community
dit_s2
File size: 1,049 Bytes
dde6c77
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "dataset_name": "imagefolder",
  "data_path": "../data/imagenet/train",
  "image_size": 256,
  "latent_size": 32,
  "latent_channels": 4,
  "num_classes": 1000,
  "vae": "mse",
  "num_dummy_samples": 1000,
  "dit_model_name": "DiT-S/2",
  "noise_rater_model_name": "DummyNoiseRater",
  "num_noise_samples": 1,
  "num_outer_noise_samples": 4,
  "diffusion_steps": 1000,
  "timestep_min_frac": 0.0,
  "timestep_max_frac": 1.0,
  "batch_size": 256,
  "train_split_ratio": 0.9,
  "inner_lr": 0.0001,
  "outer_lr": 0.001,
  "meta_steps": 100,
  "inner_steps": 1,
  "meta_refresh_steps": 10,
  "grad_clip_norm": 1.0,
  "num_inner_models": 2,
  "device": "cuda",
  "save_checkpoint": false,
  "log": true,
  "dit_lr": 0.0001,
  "dit_total_steps": 400000,
  "dit_ema_decay": 0.9999,
  "dit_log_every": 100,
  "dit_ckpt_every": 25000,
  "dit_learn_sigma": true,
  "noise_rater_checkpoint": "experiments/noise_DiT-Nano_20260222_1557_931af877/noise_rater.pt",
  "dit_init_checkpoint": null,
  "dit_resume_checkpoint": null,
  "use_ddp": true,
  "seed": 0
}