yjwsb233 commited on
Commit
30a6969
·
1 Parent(s): 012d6a6

Training in progress, epoch 1

Browse files
Files changed (4) hide show
  1. .gitignore +1 -0
  2. config.json +94 -0
  3. pytorch_model.bin +3 -0
  4. training_args.bin +3 -0
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
config.json ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/resnet-101",
3
+ "architectures": [
4
+ "ResNetForImageClassification"
5
+ ],
6
+ "depths": [
7
+ 3,
8
+ 4,
9
+ 23,
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": "beverage cans",
23
+ "1": "cardboard",
24
+ "2": "cigarette butt",
25
+ "3": "construction scrap",
26
+ "4": "electrical cables",
27
+ "5": "electronic chips",
28
+ "6": "glass",
29
+ "7": "gloves",
30
+ "8": "laptops",
31
+ "9": "masks",
32
+ "10": "medicines",
33
+ "11": "metal containers",
34
+ "12": "news paper",
35
+ "13": "other metal objects",
36
+ "14": "paper",
37
+ "15": "paper cups",
38
+ "16": "plastic bags",
39
+ "17": "plastic bottles",
40
+ "18": "plastic containers",
41
+ "19": "plastic cups",
42
+ "20": "small appliances",
43
+ "21": "smartphones",
44
+ "22": "spray cans",
45
+ "23": "syringe",
46
+ "24": "tetra pak"
47
+ },
48
+ "label2id": {
49
+ "beverage cans": 0,
50
+ "cardboard": 1,
51
+ "cigarette butt": 2,
52
+ "construction scrap": 3,
53
+ "electrical cables": 4,
54
+ "electronic chips": 5,
55
+ "glass": 6,
56
+ "gloves": 7,
57
+ "laptops": 8,
58
+ "masks": 9,
59
+ "medicines": 10,
60
+ "metal containers": 11,
61
+ "news paper": 12,
62
+ "other metal objects": 13,
63
+ "paper": 14,
64
+ "paper cups": 15,
65
+ "plastic bags": 16,
66
+ "plastic bottles": 17,
67
+ "plastic containers": 18,
68
+ "plastic cups": 19,
69
+ "small appliances": 20,
70
+ "smartphones": 21,
71
+ "spray cans": 22,
72
+ "syringe": 23,
73
+ "tetra pak": 24
74
+ },
75
+ "layer_type": "bottleneck",
76
+ "model_type": "resnet",
77
+ "num_channels": 3,
78
+ "out_features": [
79
+ "stage4"
80
+ ],
81
+ "out_indices": [
82
+ 4
83
+ ],
84
+ "problem_type": "single_label_classification",
85
+ "stage_names": [
86
+ "stem",
87
+ "stage1",
88
+ "stage2",
89
+ "stage3",
90
+ "stage4"
91
+ ],
92
+ "torch_dtype": "float32",
93
+ "transformers_version": "4.31.0.dev0"
94
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03586da3d33f605140628fc0b43414225fe3384f7704af23ed4e040141e189f8
3
+ size 170850837
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c829cf992f461cfc4894c4fddb30d00bd36cbc07277320cd4ed42b066f14da4
3
+ size 3899