Update README.md
Browse files
README.md
CHANGED
|
@@ -12,7 +12,7 @@ datasets:
|
|
| 12 |
- AIOmarRehan/Vehicles
|
| 13 |
---
|
| 14 |
|
| 15 |
-
##
|
| 16 |
|
| 17 |
This repository provides three different trained PyTorch models for vehicle image classification:
|
| 18 |
|
|
@@ -22,7 +22,7 @@ This repository provides three different trained PyTorch models for vehicle imag
|
|
| 22 |
| `cnn_model_statedict_20260226_034332.pth` | PyTorch `.pth` | **Custom CNN** trained from scratch. Baseline performance, high variance on unseen data. |
|
| 23 |
| `model.safetensors` | PyTorch `safetensors` | Lightweight **unified CNN model**, faster loading and inference, safe for sharing and reproducible deployment. |
|
| 24 |
|
| 25 |
-
###
|
| 26 |
|
| 27 |
**PyTorch `.pt` or `.pth`:**
|
| 28 |
|
|
|
|
| 12 |
- AIOmarRehan/Vehicles
|
| 13 |
---
|
| 14 |
|
| 15 |
+
## Models Included
|
| 16 |
|
| 17 |
This repository provides three different trained PyTorch models for vehicle image classification:
|
| 18 |
|
|
|
|
| 22 |
| `cnn_model_statedict_20260226_034332.pth` | PyTorch `.pth` | **Custom CNN** trained from scratch. Baseline performance, high variance on unseen data. |
|
| 23 |
| `model.safetensors` | PyTorch `safetensors` | Lightweight **unified CNN model**, faster loading and inference, safe for sharing and reproducible deployment. |
|
| 24 |
|
| 25 |
+
### How to Load
|
| 26 |
|
| 27 |
**PyTorch `.pt` or `.pth`:**
|
| 28 |
|