best quick_fid_4096=48.418213 step=150000 at 20260306-013603-UTC

.gitattributes

@@ -42,3 +42,7 @@ jepa_full_final/best/step_100000_20260305-235157-UTC/ocdbt.process_0/d/791a5a64e
 jepa_full_final/best/step_100000_20260305-235157-UTC/ocdbt.process_0/d/aaf3588a581047468ad21486c9bd8c54 filter=lfs diff=lfs merge=lfs -text
 jepa_full_final/best/step_100000_20260305-235157-UTC/ocdbt.process_0/d/c7756d4de2dfd76cf0bdfcb184da596e filter=lfs diff=lfs merge=lfs -text
 jepa_full_final/best/step_100000_20260305-235157-UTC/ocdbt.process_0/d/df1ad2e6d4e530e4504c482c8bffb276 filter=lfs diff=lfs merge=lfs -text
+jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/185f8e1627726b8615e6668d22f38117 filter=lfs diff=lfs merge=lfs -text
+jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/2a987fd68fb1ad99987498ef8b09cc68 filter=lfs diff=lfs merge=lfs -text
+jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/4c49a96d1a93b413bb569ca87828100f filter=lfs diff=lfs merge=lfs -text
+jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/c8e8ac49a896dc7fcb82f7373cb11d33 filter=lfs diff=lfs merge=lfs -text

jepa_full_final/best/step_150000_20260306-013603-UTC/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1772760944098882186, "commit_timestamp_nsecs": 1772760962871519522, "custom_metadata": {}}

jepa_full_final/best/step_150000_20260306-013603-UTC/config.json

@@ -0,0 +1,51 @@
+{
+  "mode": "jepa",
+  "data_dir": "/kaggle/input/datasets/bangchi/miniimagenet256-latents-arrayrecord-sdvae",
+  "num_classes": 100,
+  "patch_size": 2,
+  "hidden_size": 768,
+  "depth": 12,
+  "num_heads": 12,
+  "mlp_ratio": 4.0,
+  "opt": "adamw",
+  "lr": 0.0001,
+  "beta1": 0.9,
+  "beta2": 0.99,
+  "weight_decay": 0.02,
+  "global_batch": 256,
+  "steps": 300000,
+  "seed": 42,
+  "class_dropout_prob": 0.1,
+  "aug_flip_p": 0.5,
+  "aug_jitter_eps": 0.01,
+  "t_schedule": "lognormal",
+  "t_lognorm_mean": -0.4,
+  "t_lognorm_std": 1.0,
+  "lambda_jepa": 0.15,
+  "ema_decay": 0.995,
+  "mask_ratio": 0.3,
+  "student_layer": 3,
+  "teacher_layer": 7,
+  "cfg_scale": 2.0,
+  "sample_steps": 128,
+  "num_sample_images": 16,
+  "fid_n": 4096,
+  "fid_cache_path": "/kaggle/working/checkpoints_jepa/fid_real_stats_4096.npz",
+  "fid_decode_batch": 128,
+  "fid_inception_batch": 128,
+  "log_every": 500,
+  "eval_every": 5000,
+  "sample_every": 10000,
+  "fid_every": 50000,
+  "ckpt_every": 50000,
+  "run_name": "jepa_full_final",
+  "ckpt_dir": "/kaggle/working/checkpoints_jepa",
+  "best_metric": "quick_fid_4096",
+  "hf_repo_id": "",
+  "hf_username": "Bangchis",
+  "hf_repo_name": "soft-jepa-flow",
+  "hf_private": false,
+  "latent_size": 32,
+  "latent_channels": 4,
+  "vae_scaling_factor": 0.18215
+}

jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/185f8e1627726b8615e6668d22f38117

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

jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/2a987fd68fb1ad99987498ef8b09cc68

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

jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/4c49a96d1a93b413bb569ca87828100f

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

jepa_full_final/best/step_150000_20260306-013603-UTC/ocdbt.process_0/d/c8e8ac49a896dc7fcb82f7373cb11d33

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