{
  "train/loss": 2.5,
  "train/contrastive": 2.40625,
  "train/recons_loss": 0.671875,
  "train/balance_loss": 3.828125,
  "train/balance_loss_contrastive": 2.78125,
  "train/balance_loss_recons": 1.046875,
  "train/contrastive_std": 3.265625,
  "train/recons_std": 0.1513671875,
  "train/contrastive_min": 0.1162109375,
  "train/contrastive_max": 6.9375,
  "train/recons_min": 0.55859375,
  "train/recons_max": 0.96484375,
  "train/Qwen3_0.6B_layer_2": 0.96484375,
  "train/Qwen3_0.6B_layer_4": 0.6015625,
  "train/Qwen3_1.7B_layer_2": 0.58203125,
  "train/Qwen3_1.7B_layer_4": 0.69140625,
  "train/Qwen3_4B_layer_2": 0.55859375,
  "train/Qwen3_4B_layer_4": 0.625,
  "train/contrastives": null,
  "train/epoch": 1,
  "train/n_tokens": 4001792,
  "train/step": 3908
}