update model card: best quick_fid_4096=50.335258 step=100000

README.md

@@ -9,7 +9,7 @@ tags:
 - jepa
 ---
 
-# Soft-JEPA-Flow (jepa2_run)
+# Soft-JEPA-Flow (baseline_run_5)
 
 This repository stores checkpoints automatically uploaded from training.
 
@@ -25,21 +25,21 @@ This repository stores checkpoints automatically uploaded from training.
 - JEPA branch: disabled in baseline mode.
 
 ## Training Objective
-- Mode: `jepa2`
+- Mode: `baseline`
 - Objective: L_total = L_gen (rectified-flow velocity prediction objective).
-- Optimizer: `adam` (lr=`0.0001`, beta1=`0.9`, beta2=`0.99`, weight_decay=`0.06`)
-- Timestep schedule: `uniform` (mean=`-0.4`, std=`1.0`)
+- Optimizer: `adamw` (lr=`0.0001`, beta1=`0.9`, beta2=`0.99`, weight_decay=`0.0`)
+- Timestep schedule: `lognormal` (mean=`-0.4`, std=`1.0`)
 
 ## Metrics
 - Best metric key: `quick_fid_4096`
-- Best metric value: `91.158997`
-- Best checkpoint step: `50000`
+- Best metric value: `50.335258`
+- Best checkpoint step: `100000`
 
 ## Upload Timing (Automatic)
 - Upload policy: upload whenever a **new best metric** is observed.
-- Upload timestamp (UTC): `20260310-104055-UTC`
-- Run name: `jepa2_run`
-- Path for this artifact: `jepa2_run/best/step_50000_20260310-104055-UTC`
+- Upload timestamp (UTC): `20260310-113201-UTC`
+- Run name: `baseline_run_5`
+- Path for this artifact: `baseline_run_5/best/step_100000_20260310-113201-UTC`
 
 ## Evaluation Notes
 - FID pipeline uses latent decode with `stabilityai/sd-vae-ft-mse`, then InceptionV3 features.