taroii commited on
Commit
2d5e78c
1 Parent(s): 69d95b2

Upload ResNetForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +65 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/resnet-50",
3
+ "architectures": [
4
+ "ResNetForImageClassification"
5
+ ],
6
+ "depths": [
7
+ 3,
8
+ 4,
9
+ 6,
10
+ 3
11
+ ],
12
+ "downsample_in_first_stage": false,
13
+ "embedding_size": 64,
14
+ "hidden_act": "relu",
15
+ "hidden_sizes": [
16
+ 256,
17
+ 512,
18
+ 1024,
19
+ 2048
20
+ ],
21
+ "id2label": {
22
+ "0": " Renaissance",
23
+ "1": " Surrealism",
24
+ "10": " romanticism",
25
+ "2": " Unlabeled",
26
+ "3": " baroque",
27
+ "4": " cubism",
28
+ "5": " cubsim",
29
+ "6": " minimalist",
30
+ "7": " popart",
31
+ "8": " realism",
32
+ "9": " renaissance"
33
+ },
34
+ "label2id": {
35
+ " Renaissance": "0",
36
+ " Surrealism": "1",
37
+ " Unlabeled": "2",
38
+ " baroque": "3",
39
+ " cubism": "4",
40
+ " cubsim": "5",
41
+ " minimalist": "6",
42
+ " popart": "7",
43
+ " realism": "8",
44
+ " renaissance": "9",
45
+ " romanticism": "10"
46
+ },
47
+ "layer_type": "bottleneck",
48
+ "model_type": "resnet",
49
+ "num_channels": 3,
50
+ "out_features": [
51
+ "stage4"
52
+ ],
53
+ "out_indices": [
54
+ 4
55
+ ],
56
+ "stage_names": [
57
+ "stem",
58
+ "stage1",
59
+ "stage2",
60
+ "stage3",
61
+ "stage4"
62
+ ],
63
+ "torch_dtype": "float32",
64
+ "transformers_version": "4.31.0"
65
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6a96ac0f1b1e37c96ea51665b9d319fa15cc51f37c8db2a3f56e49f32f4ddce
3
+ size 94448717