Add vqvae-celeba-32-512

vqvae-celeba-32-512/ckpt/step0499999/meta.pt

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

vqvae-celeba-32-512/ckpt/step0499999/model.pt

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

vqvae-celeba-32-512/ckpt/step0499999/optimizer.pt

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

vqvae-celeba-32-512/config-2024-10-10-15-01-42.yaml

@@ -0,0 +1,40 @@
+seed: 8888
+data:
+  name: celeba
+  root: ~/data/CelebA/
+  img_size: 64
+dataloader:
+  num_workers: 4
+  pin_memory: true
+  prefetch_factor: 2
+encoder:
+  target: models.autoencoder.simple_cnn.Encoder
+  params:
+    img_channels: 3
+    hidden_dim: 256
+    n_resblocks: 2
+decoder:
+  target: models.autoencoder.simple_cnn.Decoder
+  params:
+    img_channels: 3
+    hidden_dim: 256
+    n_resblocks: 2
+quantizer:
+  target: models.quantizer.VectorQuantizer
+  params:
+    codebook_num: 512
+    codebook_dim: 32
+    l2_norm: false
+    use_ema_update: false
+    ema_decay: 0.99
+train:
+  n_steps: 500000
+  batch_size: 256
+  coef_commit: 0.25
+  print_freq: 100
+  sample_freq: 10000
+  save_freq: 10000
+  optim:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0004

vqvae-celeba-32-512/samples.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6bc00d691bb83d1188d715b0ed15990de4a677f20768cb4f331dd2a85fd19600
+size 23633089

vqvae-celeba-32-512/tensorboard/events.out.tfevents.1728543702.amax.2191947.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4189f18415e72aa5822fb1759c5fbdeb47c7c7b848bc40191546d9cf2f37d64a
+size 133917518