diffbir-mixed100-model / model_config.yaml
InsultedByMathematics's picture
Initial commit
96a88e2 verified
Raw
History Blame Contribute Delete
144 Bytes
channels: 4
depth: 18
head_dim: 64
in_dim: 128
mlp_ratio: 4.0
num_classes: 1001
num_heads: 12
patch_size: 4
vae: true
vae_scale_factor: 0.18215