Upload 4 files

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* 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
 cifar10/best-training-checkpoints/ckpt-68096000.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
+ffhq_128/best-training-checkpoints/ckpt-24064000.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

ffhq_128/best-training-checkpoints/checkpoint

@@ -0,0 +1,4 @@
+model_checkpoint_path: "ckpt-24064000"
+all_model_checkpoint_paths: "ckpt-24064000"
+all_model_checkpoint_timestamps: 1706240715.2490573
+last_preserved_timestamp: 1706240671.0879555

ffhq_128/best-training-checkpoints/ckpt-24064000.data-00000-of-00001

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

ffhq_128/config.json

@@ -0,0 +1,56 @@
+{
+    "img_size": 128,
+    "batch_size": 64,
+    "g_lr": 0.0002,
+    "g_beta1": 0.5,
+    "g_beta2": 0.99,
+    "noise_dim": 128,
+    "g_initializer": "orthogonal",
+    "g_dim": [
+        1024,
+        512,
+        256
+    ],
+    "g_heads": [
+        4,
+        4,
+        4
+    ],
+    "g_mlp": [
+        512,
+        512,
+        512
+    ],
+    "d_initializer": "orthogonal",
+    "d_enc_dim": [
+        32,
+        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,cutout",
+    "fid_batch_size": 50,
+    "gen_batch_size": 50,
+    "loss": "nsl",
+    "ema_decay": 0.999,
+    "bcr": true,
+    "cr_weight": 0.1,
+    "dec_dim": [
+        128,
+        64
+    ],
+    "n_fid_real": 2500,
+    "n_fid_gen": 2500,
+    "plot_size": 5.2,
+    "test_seed": 77
+}