Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -10,8 +10,31 @@ tags:
|
|
| 10 |
- smoke-test
|
| 11 |
---
|
| 12 |
|
| 13 |
-
# Tiny TorchScript
|
| 14 |
|
| 15 |
-
Minimal CNN traced to TorchScript for pipeline validation
|
| 16 |
-
Inputs: `[1,3,224,224]` float32. Outputs: `[1,10]` logits.
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
- smoke-test
|
| 11 |
---
|
| 12 |
|
| 13 |
+
# Tiny TorchScript CNN
|
| 14 |
|
| 15 |
+
Minimal CNN traced to TorchScript for pipeline validation and testing.
|
|
|
|
| 16 |
|
| 17 |
+
## Model Details
|
| 18 |
+
- **Architecture**: Simple 3-layer CNN (TinyCNN)
|
| 19 |
+
- **Input**: `[1, 3, 224, 224]` float32 tensor (RGB image)
|
| 20 |
+
- **Output**: `[1, 10]` logits (10 classes)
|
| 21 |
+
- **Format**: TorchScript (traced)
|
| 22 |
+
|
| 23 |
+
## Usage
|
| 24 |
+
```python
|
| 25 |
+
import torch
|
| 26 |
+
from huggingface_hub import hf_hub_download
|
| 27 |
+
|
| 28 |
+
# Download and load
|
| 29 |
+
model_path = hf_hub_download("Khauneesh/tiny-cnn-torchscript", "model.pt")
|
| 30 |
+
model = torch.jit.load(model_path, map_location="cpu")
|
| 31 |
+
model.eval()
|
| 32 |
+
|
| 33 |
+
# Inference
|
| 34 |
+
x = torch.randn(1, 3, 224, 224)
|
| 35 |
+
with torch.no_grad():
|
| 36 |
+
logits = model(x)
|
| 37 |
+
```
|
| 38 |
+
|
| 39 |
+
## Testing
|
| 40 |
+
This model is designed for smoke testing TorchScript deployment pipelines.
|