experiments/bs256_lr1e-4/validation/checkpoint_validation.json

{
  "base": {
    "source_path": "/home/nvidia/SiT-official-bs256/results-official-bs256-400k/000-SiT-S-2-Linear-velocity-None/checkpoints/4003200.pt",
    "filename": "4003200.pt",
    "bytes": 526844432,
    "sha256": "81c9246463c94a3200e7d1861a95bd923e3f3cac92e4966d861a843348142f4e",
    "metadata_fields": [
      "args",
      "ema",
      "model",
      "opt"
    ],
    "checkpoint_steps": null,
    "configured_epochs": 800,
    "global_batch_size": 256,
    "learning_rate": null,
    "model_module": "models",
    "state_dict": "ema",
    "learn_sigma": true,
    "strict_load": true,
    "forward_shape": [
      1,
      4,
      32,
      32
    ],
    "decoded_shape": [
      1,
      3,
      256,
      256
    ],
    "preview": "base_ema_euler10_seed20260815.png"
  },
  "rot-layer": {
    "source_path": "/home/nvidia/SiT-rot-layer-bs256/results-200ep/SiT-S-2-RotLayer-bs256-lr1e-4-200ep/checkpoints/4003200.pt",
    "filename": "4003200.pt",
    "bytes": 526845456,
    "sha256": "0d0a191eda63f82d6262f8b41644a3f9ceb0f3cfe66cc0accc711c8b488bbc32",
    "metadata_fields": [
      "args",
      "ema",
      "model",
      "opt"
    ],
    "checkpoint_steps": null,
    "configured_epochs": 800,
    "global_batch_size": 256,
    "learning_rate": 0.0001,
    "model_module": "models_rot_layer",
    "state_dict": "ema",
    "learn_sigma": true,
    "strict_load": true,
    "forward_shape": [
      1,
      4,
      32,
      32
    ],
    "decoded_shape": [
      1,
      3,
      256,
      256
    ],
    "preview": "rot-layer_ema_euler10_seed20260815.png"
  },
  "conv-layer": {
    "source_path": "/data/nvidia/SiT-conv-layer-bs256/results-800ep/SiT-S-2-ConvLayer-bs256-lr1e-4-800ep/checkpoints/1950000.pt",
    "filename": "1950000.pt",
    "bytes": 526985776,
    "sha256": "f3724fa2651c6fbb3a624664f057c1dd56c658d68fb356c52cf51a7684ae7548",
    "metadata_fields": [
      "args",
      "ema",
      "model",
      "opt"
    ],
    "checkpoint_steps": null,
    "configured_epochs": 800,
    "global_batch_size": 256,
    "learning_rate": 0.0001,
    "model_module": "models_conv",
    "state_dict": "ema",
    "learn_sigma": true,
    "strict_load": true,
    "forward_shape": [
      1,
      4,
      32,
      32
    ],
    "decoded_shape": [
      1,
      3,
      256,
      256
    ],
    "preview": "conv-layer_ema_euler10_seed20260815.png"
  }
}