Add model
Browse files- config.json +38 -19
- model.safetensors +2 -2
config.json
CHANGED
@@ -1,24 +1,43 @@
|
|
1 |
{
|
2 |
"architecture": "resnet50",
|
3 |
-
"
|
4 |
-
"
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
"num_classes": 5,
|
13 |
-
"optimizer": {
|
14 |
-
"betas": [
|
15 |
-
0.9,
|
16 |
-
0.999
|
17 |
],
|
18 |
-
"
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
}
|
24 |
}
|
|
|
1 |
{
|
2 |
"architecture": "resnet50",
|
3 |
+
"num_classes": 1,
|
4 |
+
"num_features": 2048,
|
5 |
+
"pretrained_cfg": {
|
6 |
+
"tag": "a1_in1k",
|
7 |
+
"custom_load": false,
|
8 |
+
"input_size": [
|
9 |
+
3,
|
10 |
+
224,
|
11 |
+
224
|
|
|
|
|
|
|
|
|
|
|
12 |
],
|
13 |
+
"test_input_size": [
|
14 |
+
3,
|
15 |
+
288,
|
16 |
+
288
|
17 |
+
],
|
18 |
+
"fixed_input_size": false,
|
19 |
+
"interpolation": "bicubic",
|
20 |
+
"crop_pct": 0.95,
|
21 |
+
"test_crop_pct": 1.0,
|
22 |
+
"crop_mode": "center",
|
23 |
+
"mean": [
|
24 |
+
0.485,
|
25 |
+
0.456,
|
26 |
+
0.406
|
27 |
+
],
|
28 |
+
"std": [
|
29 |
+
0.229,
|
30 |
+
0.224,
|
31 |
+
0.225
|
32 |
+
],
|
33 |
+
"num_classes": 1000,
|
34 |
+
"pool_size": [
|
35 |
+
7,
|
36 |
+
7
|
37 |
+
],
|
38 |
+
"first_conv": "conv1",
|
39 |
+
"classifier": "fc",
|
40 |
+
"origin_url": "https://github.com/huggingface/pytorch-image-models",
|
41 |
+
"paper_ids": "arXiv:2110.00476"
|
42 |
}
|
43 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6231e8d1a80f6a3fd2587c148f9f956e44886c2030d9378e1771eebfc3577d00
|
3 |
+
size 94281740
|