Khauneesh commited on
Commit
dc20074
·
verified ·
1 Parent(s): 54b9346

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +26 -3
README.md CHANGED
@@ -10,8 +10,31 @@ tags:
10
  - smoke-test
11
  ---
12
 
13
- # Tiny TorchScript for smoke tests
14
 
15
- Minimal CNN traced to TorchScript for pipeline validation (no conversion on server).
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.