nateraw commited on
Commit
0ecd8d3
1 Parent(s): f873581

:beers: cheers

Browse files
Files changed (1) hide show
  1. README.md +27 -0
README.md ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - image-classification
4
+ - pytorch
5
+ license: apache-2.0
6
+ datasets:
7
+ - cifar10
8
+ metrics:
9
+ - accuracy
10
+ thumbnail: https://avatars3.githubusercontent.com/u/32437151?s=460&u=4ec59abc8d21d5feea3dab323d23a5860e6996a4&v=4
11
+ ---
12
+
13
+ # Resnet50 Model from Torchvision
14
+
15
+ ## Using the model
16
+
17
+ ```
18
+ pip install modelz
19
+ ```
20
+
21
+ ```python
22
+ from modelz import ResnetModel
23
+
24
+ model = ResnetModel.from_pretrained('nateraw/resnet50')
25
+ ex_input = torch.rand(4, 3, 224, 224)
26
+ out = model(ex_input)
27
+ ```