mserrasa commited on
Commit
a7d893e
1 Parent(s): 34a1db3

End of training

Browse files
config.json CHANGED
@@ -30,41 +30,41 @@
30
  "giou_cost": 2,
31
  "giou_loss_coefficient": 2,
32
  "id2label": {
33
- "0": "No finding",
34
- "1": "Aortic enlargement",
35
- "2": "Cardiomegaly",
36
- "3": "Pleural thickening",
37
- "4": "Pulmonary fibrosis",
38
- "5": "Nodule/Mass",
39
- "6": "Lung Opacity",
40
- "7": "Pleural effusion",
41
- "8": "Other lesion",
42
- "9": "Infiltration",
43
- "10": "ILD",
44
- "11": "Calcification",
45
- "12": "Consolidation",
46
- "13": "Atelectasis",
47
- "14": "Pneumothorax"
48
  },
49
  "init_std": 0.02,
50
  "init_xavier_std": 1.0,
51
  "is_encoder_decoder": true,
52
  "label2id": {
53
- "Aortic enlargement": 1,
54
- "Atelectasis": 13,
55
- "Calcification": 11,
56
- "Cardiomegaly": 2,
57
- "Consolidation": 12,
58
- "ILD": 10,
59
- "Infiltration": 9,
60
- "Lung Opacity": 6,
61
- "No finding": 0,
62
- "Nodule/Mass": 5,
63
- "Other lesion": 8,
64
- "Pleural effusion": 7,
65
- "Pleural thickening": 3,
66
- "Pneumothorax": 14,
67
- "Pulmonary fibrosis": 4
68
  },
69
  "mask_loss_coefficient": 1,
70
  "max_position_embeddings": 1024,
 
30
  "giou_cost": 2,
31
  "giou_loss_coefficient": 2,
32
  "id2label": {
33
+ "0": "Aortic enlargement",
34
+ "1": "Atelectasis",
35
+ "2": "Calcification",
36
+ "3": "Cardiomegaly",
37
+ "4": "Consolidation",
38
+ "5": "ILD",
39
+ "6": "Infiltration",
40
+ "7": "Lung Opacity",
41
+ "8": "Nodule/Mass",
42
+ "9": "Other lesion",
43
+ "10": "Pleural effusion",
44
+ "11": "Pleural thickening",
45
+ "12": "Pneumothorax",
46
+ "13": "Pulmonary fibrosis",
47
+ "14": "No finding"
48
  },
49
  "init_std": 0.02,
50
  "init_xavier_std": 1.0,
51
  "is_encoder_decoder": true,
52
  "label2id": {
53
+ "Aortic enlargement": 0,
54
+ "Atelectasis": 1,
55
+ "Calcification": 2,
56
+ "Cardiomegaly": 3,
57
+ "Consolidation": 4,
58
+ "ILD": 5,
59
+ "Infiltration": 6,
60
+ "Lung Opacity": 7,
61
+ "No finding": 14,
62
+ "Nodule/Mass": 8,
63
+ "Other lesion": 9,
64
+ "Pleural effusion": 10,
65
+ "Pleural thickening": 11,
66
+ "Pneumothorax": 12,
67
+ "Pulmonary fibrosis": 13
68
  },
69
  "mask_loss_coefficient": 1,
70
  "max_position_embeddings": 1024,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7389e7696686bb3a208f92b1dd167154368f00a708703a1434aab32e12fc90ec
3
  size 166509224
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02a7600adf529a943977dae156cf609cfe8f557336496f6978d16c07c51e0005
3
  size 166509224
runs/Mar27_19-41-22_005f599159e0/events.out.tfevents.1711568514.005f599159e0.14033.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5920d81a2972b9316dec54846a6ab70e03eb43c5788691a3139d6f76368f757
3
+ size 6305
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:870ce42f9f2e0291757e9dd438c7064204c0cfaa5956fc64ddc10b3626ba1ae4
3
  size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84a603d8979c4acd662bb3a9aeb1ea9e82d6985542d49844baa62e109d3192ac
3
  size 4920