Add vqvae-celeba-4-512-ema

vqvae-celeba-4-512-ema/ckpt/step0499999/meta.pt

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

vqvae-celeba-4-512-ema/ckpt/step0499999/model.pt

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

vqvae-celeba-4-512-ema/ckpt/step0499999/optimizer.pt

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

vqvae-celeba-4-512-ema/config-2024-07-24-15-11-47.yaml

@@ -0,0 +1,38 @@
+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.VectorQuantizerEMA
+  params:
+    codebook_num: 512
+    codebook_dim: 4
+    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-4-512-ema/samples.zip

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

vqvae-celeba-4-512-ema/tensorboard/events.out.tfevents.1721805110.amax-ESC8000A-E11.4094170.0

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