| { | |
| "image_size": 256, | |
| "latent_dim": 16, | |
| "ae_hidden_dim": 64, | |
| "ae_num_blocks": 3, | |
| "hidden_dim": 256, | |
| "num_hidden_layers": 12, | |
| "num_heads": 8, | |
| "patch_size": 4, | |
| "text_encoder_name": "./all-MiniLM-L6-v2", | |
| "text_encoder_git_url": "https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2", | |
| "text_encoder_dim": 384, | |
| "num_timesteps": 1000, | |
| "beta_start": 0.0001, | |
| "beta_end": 0.02, | |
| "learning_rate": 0.0001, | |
| "vae_pretrain_epochs": 1, | |
| "vae_pretrain_lr": 0.001, | |
| "grad_clip_norm": 1.0, | |
| "warmup_steps": 500, | |
| "batch_size": 8, | |
| "num_epochs": 30, | |
| "dropout": 0.1, | |
| "weight_decay": 0.01, | |
| "early_stop_patience": 8, | |
| "ema_decay": 0.999, | |
| "model_name": "ShellD", | |
| "_class_name": "ShellD", | |
| "_version": "0.1" | |
| } |