model_benchmark.json

{
  "model_file": "fine_tuning_swa.keras",
  "architecture": "EfficientNetV2-S+Conv2D(256)+BN+MaxPool2D+GeMPooling+Dense(256)+BN+Dropout(0.4)+Dense(8,softmax)",
  "total_params": 23350633,
  "trainable_params": 17810225,
  "non_trainable_params": 5540408,
  "input_shape": [
    320,
    320,
    3
  ],
  "num_classes": 8,
  "model_size_mb": 226.75,
  "saved_model_size_mb": 183.29,
  "tflite_size_kb": 90483.55,
  "tfjs_size_mb": 89.54,
  "inference_ms_keras": 437.5,
  "inference_ms_tflite": 197.8,
  "metrics": {
    "train_accuracy": 0.9997,
    "val_accuracy": 0.9851,
    "test_accuracy": 0.9792,
    "test_loss": 0.3928,
    "tta_accuracy": 0.9814,
    "top1_accuracy": 0.9792,
    "top2_accuracy": 0.9918,
    "top3_accuracy": 0.9955,
    "macro_auc": 0.9975,
    "ece": 0.1813,
    "overfitting_gap": 0.0206
  }
}