Upload new config files

Author: daniel-gallo

configs/uvit_celeba_3.yaml

@@ -0,0 +1,12 @@
+model_params:
+  img_size: 64
+  patch_size: 4
+  in_chans: 3
+  embed_dim: 512
+  depth: 3
+  num_heads: 8
+  mlp_ratio: 4
+  qkv_bias: False
+  mlp_time_embed: False
+  num_classes: -1
+  normalize_timesteps: True

configs/uvit_celeba_5.yaml

@@ -0,0 +1,12 @@
+model_params:
+  img_size: 64
+  patch_size: 4
+  in_chans: 3
+  embed_dim: 512
+  depth: 5
+  num_heads: 8
+  mlp_ratio: 4
+  qkv_bias: False
+  mlp_time_embed: False
+  num_classes: -1
+  normalize_timesteps: True

configs/uvit_celeba_7.yaml

@@ -0,0 +1,12 @@
+model_params:
+  img_size: 64
+  patch_size: 4
+  in_chans: 3
+  embed_dim: 512
+  depth: 7
+  num_heads: 8
+  mlp_ratio: 4
+  qkv_bias: False
+  mlp_time_embed: False
+  num_classes: -1
+  normalize_timesteps: True

configs/uvit_cifar10_3.yaml

@@ -0,0 +1,12 @@
+model_params:
+  img_size: 32
+  patch_size: 2
+  in_chans: 3
+  embed_dim: 512
+  depth: 3
+  num_heads: 8
+  mlp_ratio: 4
+  qkv_bias: False
+  mlp_time_embed: False
+  num_classes: -1
+  normalize_timesteps: True

configs/uvit_imagenet256_3.yaml

@@ -0,0 +1,15 @@
+model_params:
+  img_size: 32
+  patch_size: 2
+  in_chans: 4
+  embed_dim: 1024
+  depth: 3
+  num_heads: 16
+  mlp_ratio: 4
+  qkv_bias: False
+  mlp_time_embed: False
+  num_classes: 1001
+  normalize_timesteps: False
+
+autoencoder:
+  autoencoder_checkpoint_path: ./checkpoints/autoencoder/autoencoder_kl.pth

configs/uvit_imagenet64_3.yaml

@@ -0,0 +1,12 @@
+model_params:
+  img_size: 64
+  patch_size: 4
+  in_chans: 3
+  embed_dim: 768
+  depth: 3
+  num_heads: 12
+  mlp_ratio: 4
+  qkv_bias: False
+  mlp_time_embed: False
+  num_classes: 1000
+  normalize_timesteps: False