Add vqvae-celeba-4-512

vqvae-celeba-4-512/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/ckpt/step0499999/model.pt

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

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

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

vqvae-celeba-4-512/config-2024-07-21-13-51-04.yaml

@@ -0,0 +1,37 @@
+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: 4
+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/samples.zip

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

vqvae-celeba-4-512/tensorboard/events.out.tfevents.1721541064.amax-ESC8000A-E11.3196102.0

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