File size: 1,108 Bytes
972362f | 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 | {
"name": "cats64",
"data_dirs": ["data/prepared/cats64"],
"runs_dir": "runs",
"checkpoints_dir": "checkpoints",
"model": {
"image_size": 64,
"in_channels": 3,
"base_channels": 64,
"channel_mults": [1, 2, 4],
"num_res_blocks": 2,
"attn_resolutions": [16],
"attention_heads": 4,
"time_dim": 256,
"text_dim": 256,
"text_layers": 4,
"text_heads": 4,
"text_ff_mult": 4,
"text_max_tokens": 24,
"text_vocab_size": 1024,
"dropout": 0.05
},
"diffusion": {
"timesteps": 1000,
"schedule": "cosine",
"pred_type": "v",
"min_snr_gamma": 5.0,
"cond_dropout": 0.1
},
"train": {
"steps": 12000,
"batch_size": 64,
"lr": 0.0002,
"weight_decay": 0.0,
"grad_clip": 1.0,
"ema_decay": 0.9995,
"warmup_steps": 500,
"log_every": 25,
"sample_every": 500,
"save_every": 500,
"keep_checkpoints": 2,
"seed": 0,
"device": "auto",
"max_images": 0,
"caption_augment": true,
"caption_dropout": 0.05,
"amp": true,
"num_sample_steps": 40,
"sample_guidance": 3.0
}
}
|