Upload transformer-vqvae-kmeans-celeba checkpoints and training logs

transformer-vqvae-kmeans-celeba/ckpt/step499999/meta.pt

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

transformer-vqvae-kmeans-celeba/ckpt/step499999/model.pt

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

transformer-vqvae-kmeans-celeba/ckpt/step499999/optimizer.pt

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

transformer-vqvae-kmeans-celeba/config-2024-02-03-09-29-17.yaml

@@ -0,0 +1,40 @@
+seed: 8888
+data:
+  target: datasets.celeba.CelebA
+  params:
+    root: ~/data/CelebA/
+    img_size: 64
+    split: train
+dataloader:
+  num_workers: 16
+  pin_memory: true
+  prefetch_factor: 2
+model:
+  target: models.transformer.Transformer
+  params:
+    codebook_num: 512
+    embed_dim: 512
+    n_heads: 8
+    n_layers: 12
+    max_tokens: 256
+vqvae:
+  target: models.vqvae.VQVAE
+  params:
+    img_channels: 3
+    hidden_dim: 256
+    n_resblocks: 2
+    codebook_num: 512
+    codebook_dim: 64
+    codebook_update: kmeans
+    ema_decay: 0.99
+  pretrained: ./runs/vqvae-kmeans-celeba/ckpt/step499999/model.pt
+train:
+  n_steps: 200000
+  batch_size: 64
+  print_freq: 100
+  sample_freq: 1000
+  save_freq: 10000
+  optim:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0001

transformer-vqvae-kmeans-celeba/config-2024-02-03-16-30-34.yaml

@@ -0,0 +1,40 @@
+seed: 8888
+data:
+  target: datasets.celeba.CelebA
+  params:
+    root: ~/data/CelebA/
+    img_size: 64
+    split: train
+dataloader:
+  num_workers: 16
+  pin_memory: true
+  prefetch_factor: 2
+model:
+  target: models.transformer.Transformer
+  params:
+    codebook_num: 512
+    embed_dim: 512
+    n_heads: 8
+    n_layers: 12
+    max_tokens: 256
+vqvae:
+  target: models.vqvae.VQVAE
+  params:
+    img_channels: 3
+    hidden_dim: 256
+    n_resblocks: 2
+    codebook_num: 512
+    codebook_dim: 64
+    codebook_update: kmeans
+    ema_decay: 0.99
+  pretrained: ./runs/vqvae-kmeans-celeba/ckpt/step499999/model.pt
+train:
+  n_steps: 500000
+  batch_size: 64
+  print_freq: 100
+  sample_freq: 1000
+  save_freq: 10000
+  optim:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0001

transformer-vqvae-kmeans-celeba/samples.zip

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

transformer-vqvae-kmeans-celeba/tensorboard/events.out.tfevents.1706923758.amax.708885.0

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

transformer-vqvae-kmeans-celeba/tensorboard/events.out.tfevents.1706949035.amax.1111906.0

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