kaczmarj commited on
Commit
3e71678
1 Parent(s): 625062f

Add files to conform to wsinfer-zoo schema

Browse files
Files changed (4) hide show
  1. config.json +1 -36
  2. model.safetensors +3 -0
  3. pytorch_model.bin +3 -0
  4. torchscript_model.pt +2 -2
config.json CHANGED
@@ -1,36 +1 @@
1
- {
2
- "architecture": "resnet34",
3
- "num_classes": 2,
4
- "class_names": [
5
- "notumor",
6
- "tumor"
7
- ],
8
- "patch_size_pixels": 350,
9
- "spacing_um_px": 0.25,
10
- "transform": [
11
- {
12
- "name": "Resize",
13
- "arguments": {
14
- "size": 224
15
- }
16
- },
17
- {
18
- "name": "ToTensor"
19
- },
20
- {
21
- "name": "Normalize",
22
- "arguments": {
23
- "mean": [
24
- 0.7238,
25
- 0.5716,
26
- 0.6779
27
- ],
28
- "std": [
29
- 0.112,
30
- 0.1459,
31
- 0.1089
32
- ]
33
- }
34
- }
35
- ]
36
- }
 
1
+ {"spec_version": "1.0", "architecture": "resnet34", "num_classes": 2, "class_names": ["notumor", "tumor"], "patch_size_pixels": 350, "spacing_um_px": 0.25, "transform": [{"name": "Resize", "arguments": {"size": 224}}, {"name": "ToTensor"}, {"name": "Normalize", "arguments": {"mean": [0.7238, 0.5716, 0.6779], "std": [0.112, 0.1459, 0.1089]}}]}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49e47bb52e60694b351a5315a4156875d42b2602adcb226f5a0fe03a5aee878c
3
+ size 85230496
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cb91ab65093080ad2adffee139db476b1c73b7371144e1bcd79be6cba80fe38
3
+ size 85282613
torchscript_model.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2829a44b26e78ec867627fa3be7516775471577d0e80059c1af8bfc79674659a
3
- size 85342598
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:663d25424a7ef7390a260c48c741e516893983dd203824f5844279fa8811c941
3
+ size 85337454