File size: 882 Bytes
a00ee36
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "experiment_name": "biggan_imagenet_lt_class_cond_res64",
  "run_setup": "local_debug",
  "deterministic_run": true,
  "num_workers": 10,

  "ddp_train": true,
  "n_nodes": 1,
  "n_gpus_per_node": 1,
  "hflips": true,
  "DA": true,
  "DiffAugment": "translation",

  "test_every": 1,
  "save_every": 1,
  "num_epochs": 3000,
  "es_patience": 50,
  "shuffle": true,

  "G_eval_mode": true,
  "ema": true,
  "use_ema": true,
  "num_G_accumulations": 1,
  "num_D_accumulations": 1,
  "num_D_steps": 1,

  "class_cond": true,
  "hier": true,
  "resolution": 64,
  "G_attn": "32",
  "D_attn": "32",
  "shared_dim": 128,
  "G_shared": true,
  "batch_size": 128,
  "D_lr": 1e-3,
  "G_lr": 1e-5,
  "G_ch": 64,
  "D_ch": 64,

  "longtail": true,
  "longtail_gen": true,
  "use_balanced_sampler": false,
  "custom_distrib_gen": false,
  "longtail_temperature": 1,

  "load_weights": ""

}