rievil commited on
Commit
41d7746
·
verified ·
1 Parent(s): 8ad0767

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +33 -3
config.json CHANGED
@@ -1,7 +1,37 @@
1
  {
2
- "architecture":"resnext101_32x8d",
3
  "num_classes": 5,
4
- "pretrained_cfg":{
5
- "input_size":[3,416,416],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  }
7
  }
 
1
  {
2
+ "architecture": "resnext101_32x8d",
3
  "num_classes": 5,
4
+ "num_features": 2048,
5
+ "pretrained_cfg": {
6
+ "tag": "tv_in1k",
7
+ "custom_load": false,
8
+ "input_size": [
9
+ 3,
10
+ 416,
11
+ 416
12
+ ],
13
+ "fixed_input_size": false,
14
+ "interpolation": "bilinear",
15
+ "crop_pct": 1,
16
+ "crop_mode": "center",
17
+ "mean": [
18
+ 0.485,
19
+ 0.456,
20
+ 0.406
21
+ ],
22
+ "std": [
23
+ 0.229,
24
+ 0.224,
25
+ 0.225
26
+ ],
27
+ "num_classes": 5,
28
+ "pool_size": [
29
+ 7,
30
+ 7
31
+ ],
32
+ "first_conv": "conv1",
33
+ "classifier": "fc",
34
+ "license": "bsd-3-clause",
35
+ "origin_url": "https://github.com/pytorch/vision"
36
  }
37
  }