ClementP commited on
Commit
8ca5a3e
1 Parent(s): 13fe0a3
Files changed (2) hide show
  1. config.json +38 -19
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,24 +1,43 @@
1
  {
2
  "architecture": "resnet50",
3
- "as_regression": true,
4
- "ema": {
5
- "decay": 0.999
6
- },
7
- "img_size": [
8
- 512,
9
- 512
10
- ],
11
- "lr": 3e-05,
12
- "num_classes": 5,
13
- "optimizer": {
14
- "betas": [
15
- 0.9,
16
- 0.999
17
  ],
18
- "weight_decay": 0.0005
19
- },
20
- "pretrained": true,
21
- "swa": {
22
- "swa_lrs": 3e-05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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:dc86e30b0f69e37e27f334d202017cbb36218bf65cc27384033e6cea73843a77
3
- size 94283660
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6231e8d1a80f6a3fd2587c148f9f956e44886c2030d9378e1771eebfc3577d00
3
+ size 94281740