update: add the sit-davae-vavae-400k stage 2 ckpt

sit-xl-1-dinov2-b-enc8-davae-vavae-400k-freq-1-coeff0.5-400k/args.json

@@ -0,0 +1,51 @@
+{
+    "output_dir": "exps",
+    "exp_name": "sit-xl-1-dinov2-b-enc8-davae-vavae-400k-freq-1-coeff0.5-400k",
+    "logging_dir": "logs",
+    "report_to": "wandb",
+    "sampling_steps": 10000,
+    "resume_step": 0,
+    "continue_train_exp_dir": null,
+    "wandb_history_path": null,
+    "model": "SiT-XL/1",
+    "num_classes": 1000,
+    "encoder_depth": 8,
+    "fused_attn": true,
+    "bn_momentum": 0.1,
+    "compile": true,
+    "qk_norm": false,
+    "use_swiglu": true,
+    "use_rope": true,
+    "use_rmsnorm": true,
+    "wo_shift": false,
+    "data_dir": "data/imagenet-latents",
+    "resolution": 256,
+    "batch_size": 256,
+    "allow_tf32": true,
+    "mixed_precision": "fp16",
+    "epochs": 1400,
+    "max_train_steps": 400000,
+    "checkpointing_steps": 50000,
+    "gradient_accumulation_steps": 1,
+    "learning_rate": 0.0001,
+    "adam_beta1": 0.9,
+    "adam_beta2": 0.999,
+    "adam_weight_decay": 0.0,
+    "adam_epsilon": 1e-08,
+    "max_grad_norm": 1.0,
+    "seed": 0,
+    "num_workers": 4,
+    "path_type": "linear",
+    "prediction": "v",
+    "cfg_prob": 0.1,
+    "enc_type": "dinov2-vit-b",
+    "proj_coeff": 0.5,
+    "weighting": "uniform",
+    "cos_loss": false,
+    "legacy": false,
+    "vae": "f16d32",
+    "vae_latents_name": "davae-vavae-align1.5-400k",
+    "vae_ckpt": "pretrained_models/davae-vavae-align1.5-400k.pt",
+    "use_sd_vae_stats": false,
+    "imagenette_stats": 0
+}

sit-xl-1-dinov2-b-enc8-davae-vavae-400k-freq-1-coeff0.5-400k/checkpoints/0400000.pt

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

sit-xl-1-dinov2-b-enc8-davae-vavae-400k-freq-1-coeff0.5-400k/sit-xl-1-dinov2-b-enc8-davae-vavae-400k-freq-1-coeff0.5-400k.csv

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