config.json

{
  "attn_dropout": 0.1,
  "class_num": 1000,
  "embed_dim_list": [
    1024,
    512,
    128
  ],
  "fractal_level": 0,
  "generator_type_list": [
    "ar",
    "ar",
    "ar"
  ],
  "grad_checkpointing": false,
  "guiding_pixel": false,
  "img_size_list": [
    64,
    4,
    1
  ],
  "label_drop_prob": 0.1,
  "num_blocks_list": [
    32,
    8,
    3
  ],
  "num_conds": 1,
  "num_heads_list": [
    16,
    8,
    4
  ],
  "proj_dropout": 0.1,
  "r_weight": 5.0
}