File size: 2,659 Bytes
0cfed93 86ac7d9 0cfed93 630e419 0cfed93 00525ec 0cfed93 00525ec 0cfed93 00525ec 0cfed93 ed91aa0 5f5853f 0cfed93 ed91aa0 86ac7d9 0cfed93 ed91aa0 0cfed93 ed91aa0 0cfed93 c187e3d |
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 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
{
"prior": {
"clip": {
"make": "openai",
"model": "ViT-L/14"
},
"net": {
"dim": 768,
"depth": 12,
"num_timesteps": 1000,
"max_text_len": 77,
"num_time_embeds": 1,
"num_image_embeds": 1,
"num_text_embeds": 1,
"dim_head": 64,
"heads": 12,
"ff_mult": 4,
"norm_out": true,
"attn_dropout": 0.05,
"ff_dropout": 0.05,
"final_proj": true,
"normformer": true,
"rotary_emb": true
},
"image_embed_dim": 768,
"image_size": 224,
"image_channels": 3,
"timesteps": 1000,
"sample_timesteps": 64,
"cond_drop_prob": 0.1,
"loss_type": "l2",
"predict_x_start": true,
"beta_schedule": "cosine",
"condition_on_text_encodings": true
},
"data": {
"batch_size": 220,
"num_data_points": 500000000,
"eval_every_seconds": 1800,
"image_url": "s3://s-laion5b/embeddings/vit-l-14/laion2B-en/img_emb/",
"meta_url": "s3://s-laion5b/embeddings/vit-l-14/laion2B-en/metadata/",
"splits": {
"train": 0.9,
"val": 6e-5,
"test":0.09994000000000003
}
},
"train": {
"epochs": 5,
"lr": 1.1e-4,
"wd": 6.02e-2,
"max_grad_norm": 0.5,
"use_ema": true,
"ema_beta": 0.9999,
"ema_update_after_step": 50,
"warmup_steps": 50,
"amp": false,
"save_every_seconds": 3600,
"eval_timesteps": [64, 1000],
"random_seed": 84513
},
"tracker": {
"data_path": ".prior",
"overwrite_data_path": true,
"log": {
"log_type": "wandb",
"wandb_entity": "nousr_laion",
"wandb_project": "dalle2_diffusion_prior",
"wandb_resume": false,
"verbose": true
},
"load": {
"resume": false
},
"save": [
{
"save_to": "local",
"save_type": "checkpoint",
"save_latest_to": ".prior/latest_checkpoint.pth",
"save_best_to": ".prior/best_checkpoint.pth"
},
{
"save_to": "huggingface",
"huggingface_repo": "laion/DALLE2-PyTorch",
"save_meta_to": "prior/",
"save_latest_to": "prior/latest.pth",
"save_best_to": "prior/best.pth",
"save_type": "model"
}
]
}
}
|