Upload vqvae-celeba checkpoints and training logs

vqvae-celeba/ckpt/step499999/meta.pt

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

vqvae-celeba/ckpt/step499999/model.pt

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

vqvae-celeba/ckpt/step499999/optimizer.pt

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

vqvae-celeba/config-2024-01-29-14-54-33.yaml

@@ -0,0 +1,38 @@
+seed: 8888
+data:
+  target: datasets.celeba.CelebA
+  params:
+    root: ~/data/CelebA/
+    img_size: 64
+    split: train
+data_valid:
+  target: datasets.celeba.CelebA
+  params:
+    root: ~/data/CelebA/
+    img_size: 64
+    split: valid
+dataloader:
+  num_workers: 16
+  pin_memory: true
+  prefetch_factor: 2
+model:
+  target: models.vqvae.VQVAE
+  params:
+    img_channels: 3
+    hidden_dim: 256
+    n_resblocks: 2
+    codebook_num: 512
+    codebook_dim: 64
+    codebook_update: learned
+    ema_decay: 0.99
+train:
+  n_steps: 250000
+  batch_size: 256
+  coef_commit: 0.25
+  print_freq: 100
+  sample_freq: 1000
+  save_freq: 10000
+  optim:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0004

vqvae-celeba/config-2024-01-30-03-24-30.yaml

@@ -0,0 +1,38 @@
+seed: 8888
+data:
+  target: datasets.celeba.CelebA
+  params:
+    root: ~/data/CelebA/
+    img_size: 64
+    split: train
+data_valid:
+  target: datasets.celeba.CelebA
+  params:
+    root: ~/data/CelebA/
+    img_size: 64
+    split: valid
+dataloader:
+  num_workers: 16
+  pin_memory: true
+  prefetch_factor: 2
+model:
+  target: models.vqvae.VQVAE
+  params:
+    img_channels: 3
+    hidden_dim: 256
+    n_resblocks: 2
+    codebook_num: 512
+    codebook_dim: 64
+    codebook_update: learned
+    ema_decay: 0.99
+train:
+  n_steps: 500000
+  batch_size: 256
+  coef_commit: 0.25
+  print_freq: 100
+  sample_freq: 1000
+  save_freq: 10000
+  optim:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0004

vqvae-celeba/samples.zip

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

vqvae-celeba/tensorboard/events.out.tfevents.1706540073.jason-system.24915.0

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

vqvae-celeba/tensorboard/events.out.tfevents.1706585070.jason-system.53930.0

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