{
  "format": "aegis-mlx-npz-v2",
  "num_tensors": 320,
  "keys": [
    "conv1__weight",
    "bn1__weight",
    "bn1__bias",
    "bn1__running_mean",
    "bn1__running_var",
    "bn1__num_batches_tracked",
    "layer1__0__conv1__weight",
    "layer1__0__bn1__weight",
    "layer1__0__bn1__bias",
    "layer1__0__bn1__running_mean",
    "layer1__0__bn1__running_var",
    "layer1__0__bn1__num_batches_tracked",
    "layer1__0__conv2__weight",
    "layer1__0__bn2__weight",
    "layer1__0__bn2__bias",
    "layer1__0__bn2__running_mean",
    "layer1__0__bn2__running_var",
    "layer1__0__bn2__num_batches_tracked",
    "layer1__0__conv3__weight",
    "layer1__0__bn3__weight",
    "layer1__0__bn3__bias",
    "layer1__0__bn3__running_mean",
    "layer1__0__bn3__running_var",
    "layer1__0__bn3__num_batches_tracked",
    "layer1__0__downsample__0__weight",
    "layer1__0__downsample__1__weight",
    "layer1__0__downsample__1__bias",
    "layer1__0__downsample__1__running_mean",
    "layer1__0__downsample__1__running_var",
    "layer1__0__downsample__1__num_batches_tracked",
    "layer1__1__conv1__weight",
    "layer1__1__bn1__weight",
    "layer1__1__bn1__bias",
    "layer1__1__bn1__running_mean",
    "layer1__1__bn1__running_var",
    "layer1__1__bn1__num_batches_tracked",
    "layer1__1__conv2__weight",
    "layer1__1__bn2__weight",
    "layer1__1__bn2__bias",
    "layer1__1__bn2__running_mean",
    "layer1__1__bn2__running_var",
    "layer1__1__bn2__num_batches_tracked",
    "layer1__1__conv3__weight",
    "layer1__1__bn3__weight",
    "layer1__1__bn3__bias",
    "layer1__1__bn3__running_mean",
    "layer1__1__bn3__running_var",
    "layer1__1__bn3__num_batches_tracked",
    "layer1__2__conv1__weight",
    "layer1__2__bn1__weight",
    "layer1__2__bn1__bias",
    "layer1__2__bn1__running_mean",
    "layer1__2__bn1__running_var",
    "layer1__2__bn1__num_batches_tracked",
    "layer1__2__conv2__weight",
    "layer1__2__bn2__weight",
    "layer1__2__bn2__bias",
    "layer1__2__bn2__running_mean",
    "layer1__2__bn2__running_var",
    "layer1__2__bn2__num_batches_tracked",
    "layer1__2__conv3__weight",
    "layer1__2__bn3__weight",
    "layer1__2__bn3__bias",
    "layer1__2__bn3__running_mean",
    "layer1__2__bn3__running_var",
    "layer1__2__bn3__num_batches_tracked",
    "layer2__0__conv1__weight",
    "layer2__0__bn1__weight",
    "layer2__0__bn1__bias",
    "layer2__0__bn1__running_mean",
    "layer2__0__bn1__running_var",
    "layer2__0__bn1__num_batches_tracked",
    "layer2__0__conv2__weight",
    "layer2__0__bn2__weight",
    "layer2__0__bn2__bias",
    "layer2__0__bn2__running_mean",
    "layer2__0__bn2__running_var",
    "layer2__0__bn2__num_batches_tracked",
    "layer2__0__conv3__weight",
    "layer2__0__bn3__weight",
    "layer2__0__bn3__bias",
    "layer2__0__bn3__running_mean",
    "layer2__0__bn3__running_var",
    "layer2__0__bn3__num_batches_tracked",
    "layer2__0__downsample__0__weight",
    "layer2__0__downsample__1__weight",
    "layer2__0__downsample__1__bias",
    "layer2__0__downsample__1__running_mean",
    "layer2__0__downsample__1__running_var",
    "layer2__0__downsample__1__num_batches_tracked",
    "layer2__1__conv1__weight",
    "layer2__1__bn1__weight",
    "layer2__1__bn1__bias",
    "layer2__1__bn1__running_mean",
    "layer2__1__bn1__running_var",
    "layer2__1__bn1__num_batches_tracked",
    "layer2__1__conv2__weight",
    "layer2__1__bn2__weight",
    "layer2__1__bn2__bias",
    "layer2__1__bn2__running_mean",
    "layer2__1__bn2__running_var",
    "layer2__1__bn2__num_batches_tracked",
    "layer2__1__conv3__weight",
    "layer2__1__bn3__weight",
    "layer2__1__bn3__bias",
    "layer2__1__bn3__running_mean",
    "layer2__1__bn3__running_var",
    "layer2__1__bn3__num_batches_tracked",
    "layer2__2__conv1__weight",
    "layer2__2__bn1__weight",
    "layer2__2__bn1__bias",
    "layer2__2__bn1__running_mean",
    "layer2__2__bn1__running_var",
    "layer2__2__bn1__num_batches_tracked",
    "layer2__2__conv2__weight",
    "layer2__2__bn2__weight",
    "layer2__2__bn2__bias",
    "layer2__2__bn2__running_mean",
    "layer2__2__bn2__running_var",
    "layer2__2__bn2__num_batches_tracked",
    "layer2__2__conv3__weight",
    "layer2__2__bn3__weight",
    "layer2__2__bn3__bias",
    "layer2__2__bn3__running_mean",
    "layer2__2__bn3__running_var",
    "layer2__2__bn3__num_batches_tracked",
    "layer2__3__conv1__weight",
    "layer2__3__bn1__weight",
    "layer2__3__bn1__bias",
    "layer2__3__bn1__running_mean",
    "layer2__3__bn1__running_var",
    "layer2__3__bn1__num_batches_tracked",
    "layer2__3__conv2__weight",
    "layer2__3__bn2__weight",
    "layer2__3__bn2__bias",
    "layer2__3__bn2__running_mean",
    "layer2__3__bn2__running_var",
    "layer2__3__bn2__num_batches_tracked",
    "layer2__3__conv3__weight",
    "layer2__3__bn3__weight",
    "layer2__3__bn3__bias",
    "layer2__3__bn3__running_mean",
    "layer2__3__bn3__running_var",
    "layer2__3__bn3__num_batches_tracked",
    "layer3__0__conv1__weight",
    "layer3__0__bn1__weight",
    "layer3__0__bn1__bias",
    "layer3__0__bn1__running_mean",
    "layer3__0__bn1__running_var",
    "layer3__0__bn1__num_batches_tracked",
    "layer3__0__conv2__weight",
    "layer3__0__bn2__weight",
    "layer3__0__bn2__bias",
    "layer3__0__bn2__running_mean",
    "layer3__0__bn2__running_var",
    "layer3__0__bn2__num_batches_tracked",
    "layer3__0__conv3__weight",
    "layer3__0__bn3__weight",
    "layer3__0__bn3__bias",
    "layer3__0__bn3__running_mean",
    "layer3__0__bn3__running_var",
    "layer3__0__bn3__num_batches_tracked",
    "layer3__0__downsample__0__weight",
    "layer3__0__downsample__1__weight",
    "layer3__0__downsample__1__bias",
    "layer3__0__downsample__1__running_mean",
    "layer3__0__downsample__1__running_var",
    "layer3__0__downsample__1__num_batches_tracked",
    "layer3__1__conv1__weight",
    "layer3__1__bn1__weight",
    "layer3__1__bn1__bias",
    "layer3__1__bn1__running_mean",
    "layer3__1__bn1__running_var",
    "layer3__1__bn1__num_batches_tracked",
    "layer3__1__conv2__weight",
    "layer3__1__bn2__weight",
    "layer3__1__bn2__bias",
    "layer3__1__bn2__running_mean",
    "layer3__1__bn2__running_var",
    "layer3__1__bn2__num_batches_tracked",
    "layer3__1__conv3__weight",
    "layer3__1__bn3__weight",
    "layer3__1__bn3__bias",
    "layer3__1__bn3__running_mean",
    "layer3__1__bn3__running_var",
    "layer3__1__bn3__num_batches_tracked",
    "layer3__2__conv1__weight",
    "layer3__2__bn1__weight",
    "layer3__2__bn1__bias",
    "layer3__2__bn1__running_mean",
    "layer3__2__bn1__running_var",
    "layer3__2__bn1__num_batches_tracked",
    "layer3__2__conv2__weight",
    "layer3__2__bn2__weight",
    "layer3__2__bn2__bias",
    "layer3__2__bn2__running_mean",
    "layer3__2__bn2__running_var",
    "layer3__2__bn2__num_batches_tracked",
    "layer3__2__conv3__weight",
    "layer3__2__bn3__weight",
    "layer3__2__bn3__bias",
    "layer3__2__bn3__running_mean",
    "layer3__2__bn3__running_var",
    "layer3__2__bn3__num_batches_tracked",
    "layer3__3__conv1__weight",
    "layer3__3__bn1__weight",
    "layer3__3__bn1__bias",
    "layer3__3__bn1__running_mean",
    "layer3__3__bn1__running_var",
    "layer3__3__bn1__num_batches_tracked",
    "layer3__3__conv2__weight",
    "layer3__3__bn2__weight",
    "layer3__3__bn2__bias",
    "layer3__3__bn2__running_mean",
    "layer3__3__bn2__running_var",
    "layer3__3__bn2__num_batches_tracked",
    "layer3__3__conv3__weight",
    "layer3__3__bn3__weight",
    "layer3__3__bn3__bias",
    "layer3__3__bn3__running_mean",
    "layer3__3__bn3__running_var",
    "layer3__3__bn3__num_batches_tracked",
    "layer3__4__conv1__weight",
    "layer3__4__bn1__weight",
    "layer3__4__bn1__bias",
    "layer3__4__bn1__running_mean",
    "layer3__4__bn1__running_var",
    "layer3__4__bn1__num_batches_tracked",
    "layer3__4__conv2__weight",
    "layer3__4__bn2__weight",
    "layer3__4__bn2__bias",
    "layer3__4__bn2__running_mean",
    "layer3__4__bn2__running_var",
    "layer3__4__bn2__num_batches_tracked",
    "layer3__4__conv3__weight",
    "layer3__4__bn3__weight",
    "layer3__4__bn3__bias",
    "layer3__4__bn3__running_mean",
    "layer3__4__bn3__running_var",
    "layer3__4__bn3__num_batches_tracked",
    "layer3__5__conv1__weight",
    "layer3__5__bn1__weight",
    "layer3__5__bn1__bias",
    "layer3__5__bn1__running_mean",
    "layer3__5__bn1__running_var",
    "layer3__5__bn1__num_batches_tracked",
    "layer3__5__conv2__weight",
    "layer3__5__bn2__weight",
    "layer3__5__bn2__bias",
    "layer3__5__bn2__running_mean",
    "layer3__5__bn2__running_var",
    "layer3__5__bn2__num_batches_tracked",
    "layer3__5__conv3__weight",
    "layer3__5__bn3__weight",
    "layer3__5__bn3__bias",
    "layer3__5__bn3__running_mean",
    "layer3__5__bn3__running_var",
    "layer3__5__bn3__num_batches_tracked",
    "layer4__0__conv1__weight",
    "layer4__0__bn1__weight",
    "layer4__0__bn1__bias",
    "layer4__0__bn1__running_mean",
    "layer4__0__bn1__running_var",
    "layer4__0__bn1__num_batches_tracked",
    "layer4__0__conv2__weight",
    "layer4__0__bn2__weight",
    "layer4__0__bn2__bias",
    "layer4__0__bn2__running_mean",
    "layer4__0__bn2__running_var",
    "layer4__0__bn2__num_batches_tracked",
    "layer4__0__conv3__weight",
    "layer4__0__bn3__weight",
    "layer4__0__bn3__bias",
    "layer4__0__bn3__running_mean",
    "layer4__0__bn3__running_var",
    "layer4__0__bn3__num_batches_tracked",
    "layer4__0__downsample__0__weight",
    "layer4__0__downsample__1__weight",
    "layer4__0__downsample__1__bias",
    "layer4__0__downsample__1__running_mean",
    "layer4__0__downsample__1__running_var",
    "layer4__0__downsample__1__num_batches_tracked",
    "layer4__1__conv1__weight",
    "layer4__1__bn1__weight",
    "layer4__1__bn1__bias",
    "layer4__1__bn1__running_mean",
    "layer4__1__bn1__running_var",
    "layer4__1__bn1__num_batches_tracked",
    "layer4__1__conv2__weight",
    "layer4__1__bn2__weight",
    "layer4__1__bn2__bias",
    "layer4__1__bn2__running_mean",
    "layer4__1__bn2__running_var",
    "layer4__1__bn2__num_batches_tracked",
    "layer4__1__conv3__weight",
    "layer4__1__bn3__weight",
    "layer4__1__bn3__bias",
    "layer4__1__bn3__running_mean",
    "layer4__1__bn3__running_var",
    "layer4__1__bn3__num_batches_tracked",
    "layer4__2__conv1__weight",
    "layer4__2__bn1__weight",
    "layer4__2__bn1__bias",
    "layer4__2__bn1__running_mean",
    "layer4__2__bn1__running_var",
    "layer4__2__bn1__num_batches_tracked",
    "layer4__2__conv2__weight",
    "layer4__2__bn2__weight",
    "layer4__2__bn2__bias",
    "layer4__2__bn2__running_mean",
    "layer4__2__bn2__running_var",
    "layer4__2__bn2__num_batches_tracked",
    "layer4__2__conv3__weight",
    "layer4__2__bn3__weight",
    "layer4__2__bn3__bias",
    "layer4__2__bn3__running_mean",
    "layer4__2__bn3__running_var",
    "layer4__2__bn3__num_batches_tracked",
    "fc__weight",
    "fc__bias"
  ],
  "meta": {
    "stage": 2,
    "model_name": "resnet50",
    "num_classes": 9,
    "checkpoint": "models\\stage2\\artifacts\\checkpoint_best.pt",
    "checkpoint_resolved": "V:\\@personal\\aegis\\models\\stage2\\artifacts\\checkpoint_best.pt",
    "metrics": {
      "accuracy": 0.9951397326852977,
      "balanced_accuracy": 0.9933170561562609,
      "macro_f1": 0.9933170561562609,
      "weighted_f1": 0.9951397326852977,
      "train_loss": 0.009842748337929355
    }
  }
}