Hugging Face's logo

UT-Austin-PML
/
BetaDiffusion

Model card Files Community
BetaDiffusion
File size: 1,705 Bytes 
bccc6c4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
{
  "dataset_kwargs": {
    "class_name": "training.dataset.ImageFolderDataset",
    "path": "datasets/cifar10-32x32.zip",
    "use_labels": false,
    "xflip": false,
    "cache": true,
    "resolution": 32,
    "max_size": 50000
  },
  "data_loader_kwargs": {
    "pin_memory": true,
    "num_workers": 1,
    "prefetch_factor": 2
  },
  "network_kwargs": {
    "model_type": "SongUNet",
    "embedding_type": "positional",
    "encoder_type": "standard",
    "decoder_type": "standard",
    "channel_mult_noise": 1,
    "resample_filter": [
      1,
      1
    ],
    "model_channels": 128,
    "channel_mult": [
      2,
      2,
      2
    ],
    "class_name": "training.networks.BetaDiffPrecond",
    "augment_dim": 9,
    "dropout": 0.13,
    "use_fp16": false
  },
  "loss_kwargs": {
    "class_name": "training.loss.BetaDiffLoss",
    "eta": 10000.0,
    "sigmoid_start": 10.0,
    "sigmoid_end": -13.0,
    "sigmoid_power": 1.0,
    "Scale": 0.39,
    "Shift": 0.6,
    "lossType": "KLUB_AS"
  },
  "optimizer_kwargs": {
    "class_name": "torch.optim.Adam",
    "lr": 0.0002,
    "betas": [
      0.9,
      0.999
    ],
    "eps": 1e-08
  },
  "augment_kwargs": {
    "class_name": "training.augment.AugmentPipe",
    "p": 0.12,
    "xflip": 100000000.0,
    "yflip": 1,
    "scale": 1,
    "rotate_frac": 1,
    "aniso": 1,
    "translate_frac": 1
  },
  "total_kimg": 200000,
  "ema_halflife_kimg": 500,
  "batch_size": 288,
  "batch_gpu": null,
  "loss_scaling": 1.0,
  "cudnn_benchmark": true,
  "kimg_per_tick": 50,
  "snapshot_ticks": 50,
  "state_dump_ticks": 500,
  "seed": 17439077,
  "run_dir": "betadiff-train-runs/00018-cifar10-32x32-uncond-ddpmpp-betadiff-gpus8-batch288-fp32"
}