update: upload joint-train intermediate ckpt

sit-xl-1-dinov2-b-enc8-invae-freq1-lr1e-4-momentum0.1-l1-lpips-kl-gan-vaealign1.5-unfreezedecoder-compile-400k/args.json

@@ -0,0 +1,61 @@
+{
+    "output_dir": "exps",
+    "exp_name": "sit-xl-1-dinov2-b-enc8-invae-freq1-lr1e-4-momentum0.1-l1-lpips-kl-gan-vaealign1.5-unfreezedecoder-compile-400k",
+    "logging_dir": "logs",
+    "report_to": "wandb",
+    "sampling_steps": 10000,
+    "resume_step": 0,
+    "continue_train_exp_dir": null,
+    "wandb_history_path": null,
+    "requeue": false,
+    "model": "SiT-XL/1",
+    "num_classes": 1000,
+    "encoder_depth": 8,
+    "fused_attn": true,
+    "qk_norm": false,
+    "bn_momentum": 0.1,
+    "bn_affine": false,
+    "compile": true,
+    "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",
+    "legacy": false,
+    "vae": "f16d32",
+    "vae_ckpt": "pretrained_models/invae-f16d32.pt",
+    "use_sd_vae_stats": false,
+    "vae_repa_align": false,
+    "freeze_bn_stats": false,
+    "vae_lora": false,
+    "vae_lora_rank": 64,
+    "vae_lora_alpha": 16,
+    "vae_lora_dropout": 0.0,
+    "vae_lora_bias": "none",
+    "disc_pretrained_ckpt": "pretrained_models/invae-discriminator-ckpt.pt",
+    "loss_cfg_path": "configs/loss_cfg/l1_lpips_kl_gan.yaml",
+    "vae_learning_rate": 0.0001,
+    "disc_learning_rate": 0.0001,
+    "vae_train_freq": 1,
+    "unfreeze_decoder": true,
+    "vae_align": true,
+    "vae_align_proj_coeff": 1.5
+}

sit-xl-1-dinov2-b-enc8-invae-freq1-lr1e-4-momentum0.1-l1-lpips-kl-gan-vaealign1.5-unfreezedecoder-compile-400k/checkpoints/0400000.pt

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

sit-xl-1-dinov2-b-enc8-invae-freq1-lr1e-4-momentum0.1-l1-lpips-kl-gan-vaealign1.5-unfreezedecoder-compile-400k/sit-xl-1-dinov2-b-enc8-invae-freq1-lr1e-4-momentum0.1-l1-lpips-kl-gan-vaealign1.5-unfreezedecoder-compile-400k.csv

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