checkpoints/checkpoint-epoch-6.json

{
  "epoch": 6,
  "step": null,
  "metrics": {
    "train_loss": 0.05330577236082819,
    "fid_score": 411.45465087890625,
    "best_fid": 411.45465087890625
  },
  "timestamp": "2025-12-03T10:31:09.772849",
  "hidden_size": 768,
  "num_heads": 12,
  "expansion": 2.5,
  "vocab_size": 50257,
  "max_text_len": 77,
  "frames": 16,
  "height": 64,
  "width": 64,
  "max_steps": 4,
  "patch_size": [
    2,
    8,
    8
  ],
  "text_encoder_layers": 8,
  "hrm_blocks": 3,
  "video_seq_len": 512
}