File size: 741 Bytes
8321f11 c7a4a3a 8321f11 693d0ab 8321f11 | 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 | {
"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"
} |