Skip to content

Add v1.1 GPU-trained ResNet50 baseline#1

Merged
rhouhou merged 4 commits into
mainfrom
v1.1-gpu-baseline
Jul 2, 2026
Merged

Add v1.1 GPU-trained ResNet50 baseline#1
rhouhou merged 4 commits into
mainfrom
v1.1-gpu-baseline

Conversation

@rhouhou

@rhouhou rhouhou commented Jul 2, 2026

Copy link
Copy Markdown
Owner

Summary

This PR adds the v1.1 GPU-trained ResNet50 baseline.

Updates

  • Added Colab GPU training configuration
  • Updated test metrics using a GPU-trained ResNet50 model
  • Updated ROC and PR curves
  • Updated confusion matrix
  • Updated Grad-CAM examples
  • Updated README and MODEL_CARD with v1.1 results

Results

Model Accuracy ROC-AUC PR-AUC Brier Score
ResNet50 GPU baseline 0.902 0.971 0.975 0.078

Notes

The dataset, PyTorch checkpoint, and ONNX model are not included in the repository.

@rhouhou rhouhou merged commit b677c28 into main Jul 2, 2026
2 checks passed
@rhouhou rhouhou deleted the v1.1-gpu-baseline branch July 2, 2026 09:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant