Skip to content

Add PyTorch CNN image classification tutorial notebook#70

Merged
nee1k merged 1 commit intomainfrom
feature/pytorch-cnn-tutorial-notebook
Feb 3, 2026
Merged

Add PyTorch CNN image classification tutorial notebook#70
nee1k merged 1 commit intomainfrom
feature/pytorch-cnn-tutorial-notebook

Conversation

@nee1k
Copy link
Copy Markdown
Collaborator

@nee1k nee1k commented Feb 3, 2026

Summary

Adds a new tutorial notebook for PyTorch CNN image classification, aligned with the structure of GettingStarted.ipynb.

Changes

  • New notebook: examples/notebooks/PyTorch_CNN_Image_Classification.ipynb
  • Uses CIFAR-10 with a subset of 10k training samples and 5 epochs for faster training
  • CNN architecture: 3 conv blocks (BatchNorm, ReLU, MaxPool), FC layers, dropout
  • Full Patra Model Card workflow: ModelCard metadata, AIModel, requirements, validate, save, and optional submit
  • Includes inference example and optional submission to Patra server (card only or with model/artifacts)

Rationale

Provides a parallel tutorial path for PyTorch + image classification (vs. TensorFlow + tabular in GettingStarted), improving discoverability and reproducibility documentation for CNN models.

- New tutorial notebook aligned with GettingStarted.ipynb structure
- CIFAR-10 dataset with subset (10k samples) and 5 epochs for faster training
- CNN architecture with conv blocks, batch norm, dropout
- Full Patra Model Card workflow: metadata, AIModel, requirements, save/submit
- Inference example and optional submission to Patra server
@nee1k nee1k merged commit 0d3803e into main Feb 3, 2026
2 checks passed
@nee1k nee1k deleted the feature/pytorch-cnn-tutorial-notebook branch February 3, 2026 01:20
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