add vq-f8-n256

vq-f8-n256/config.yaml

@@ -0,0 +1,48 @@
+model:
+  base_learning_rate: 4.5e-06
+  target: ldm.models.autoencoder.VQModel
+  params:
+    embed_dim: 4
+    n_embed: 256
+    monitor: val/rec_loss
+    ddconfig:
+      double_z: false
+      z_channels: 4
+      resolution: 256
+      in_channels: 3
+      out_ch: 3
+      ch: 128
+      ch_mult:
+      - 1
+      - 2
+      - 2
+      - 4
+      num_res_blocks: 2
+      attn_resolutions:
+      - 32
+      dropout: 0.0
+    lossconfig:
+      target: taming.modules.losses.vqperceptual.VQLPIPSWithDiscriminator
+      params:
+        disc_conditional: false
+        disc_in_channels: 3
+        disc_start: 250001
+        disc_weight: 0.75
+        codebook_weight: 1.0
+
+data:
+  target: main.DataModuleFromConfig
+  params:
+    batch_size: 10
+    num_workers: 20
+    wrap: true
+    train:
+      target: ldm.data.openimages.FullOpenImagesTrain
+      params:
+        size: 384
+        crop_size: 256
+    validation:
+      target: ldm.data.openimages.FullOpenImagesValidation
+      params:
+        size: 384
+        crop_size: 256

vq-f8-n256/model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97c5c2dd3e8eb23f56163b4006ebcee21991e9b4f2b40fdcd2cb65cb074f3e65
+size 904232794