stage-2 / mlx /stage2_weights.json
RanveerSoni98's picture
Upload folder using huggingface_hub
76d70f2 verified
Raw History Blame Contribute Delete
11.9 kB
{
"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
}
}
}