Upload 4 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+celeba_64/best-training-checkpoints/ckpt-72192000.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

celeba_64/best-training-checkpoints/checkpoint

@@ -0,0 +1,4 @@
+model_checkpoint_path: "ckpt-72192000"
+all_model_checkpoint_paths: "ckpt-72192000"
+all_model_checkpoint_timestamps: 1692408354.6423366
+last_preserved_timestamp: 1692231165.9002016

celeba_64/best-training-checkpoints/ckpt-72192000.data-00000-of-00001

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

celeba_64/config.json

@@ -0,0 +1,49 @@
+{
+    "img_size": 64,
+    "batch_size": 128,
+    "g_lr": 0.0002,
+    "g_beta1": 0.5,
+    "g_beta2": 0.99,
+    "noise_dim": 128,
+    "g_initializer": "orthogonal",
+    "g_dim": [
+        1024,
+        256,
+        64
+    ],
+    "g_heads": [
+        4,
+        4,
+        4
+    ],
+    "g_mlp": [
+        512,
+        512,
+        512
+    ],
+    "d_initializer": "orthogonal",
+    "d_enc_dim": [
+        64,
+        128,
+        256
+    ],
+    "d_out_dim": [
+        512,
+        1024
+    ],
+    "d_heads": 4,
+    "d_mlp": 512,
+    "d_lr": 0.0002,
+    "d_beta1": 0.5,
+    "d_beta2": 0.99,
+    "gp_weight": 0.0001,
+    "policy": "color,translation",
+    "fid_batch_size": 50,
+    "gen_batch_size": 50,
+    "loss": "nsl",
+    "ema_decay": 0.999,
+    "n_fid_real": 2500,
+    "n_fid_gen": 2500,
+    "plot_size": 5.2,
+    "test_seed": 77
+}