Hugging Face's logo

daidedou
/
diffumatch_model

Model card Files
xet
Community
diffumatch_model
File size: 1,109 Bytes 
11b952d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
{
  "misc": {
    "cuda": true,
    "device": 0,
    "checkpoint_interval": 1,
    "log_interval": 812,
    "desc": null,
    "precond": false,
    "dry_run": false
  },
  "data": {
    "root_dir": "dfaust_fmap_30x30_clean.npy",
    "name": "DFAUST_fmap_30",
    "n_fmap": 30,
    "out": "fmap_exps",
    "cond": false,
    "in_memory": "RAM",
    "normalize": true,
    "abs": true,
    "run": "fmap_exps/00053-DFAUST_fmap_30abs-uncond-DiT-gpus1-batch256-fp32"
  },
  "add_name": {
    "do": false,
    "name": "bis"
  },
  "architecture": {
    "model": "DiT",
    "name_arch": "DiT-S/4",
    "input_type": "img",
    "cond": false
  },
  "hyper_params": {
    "iterations": 200,
    "batch_size": 256,
    "lr": 0.001,
    "lr_rampup_kimg": 10000,
    "ema_halflife_nshape": 500,
    "ema_rampup_ratio": 0.05,
    "dropout": 0,
    "loss_name": "VPLoss",
    "ls": 1
  },
  "perfs": {
    "fp16": false,
    "workers": 1
  },
  "resume": {
    "pkl": null,
    "transfer": null,
    "kimg_per_tick": 5,
    "snapshot_ticks": 50,
    "state_dump_ticks": 50
  },
  "datadir": "./data",
  "device": "cuda:0"
}