Upload 4 files

.gitattributes

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

cifar10/best-training-checkpoints/checkpoint

@@ -0,0 +1,4 @@
+model_checkpoint_path: "ckpt-68096000"
+all_model_checkpoint_paths: "ckpt-68096000"
+all_model_checkpoint_timestamps: 1706237675.9763436
+last_preserved_timestamp: 1706237671.1954017

cifar10/best-training-checkpoints/ckpt-68096000.data-00000-of-00001

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

cifar10/config.json

@@ -0,0 +1,51 @@
+{
+    "img_size": 32,
+    "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": [
+        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,
+    "bcr": false,
+    "cr_weight": 0.1,
+    "dec_dim": false,
+    "n_fid_real": 2500,
+    "n_fid_gen": 2500,
+    "plot_size": 5.2,
+    "test_seed": 77
+}