End of training
Browse files
config.json
CHANGED
@@ -31,40 +31,22 @@
|
|
31 |
"giou_loss_coefficient": 2,
|
32 |
"id2label": {
|
33 |
"0": "Aortic enlargement",
|
34 |
-
"1": "
|
35 |
-
"2": "
|
36 |
-
"3": "
|
37 |
-
"4": "
|
38 |
-
"5": "
|
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 |
-
"
|
55 |
-
"
|
56 |
-
"
|
57 |
-
"
|
58 |
-
"
|
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,
|
|
|
31 |
"giou_loss_coefficient": 2,
|
32 |
"id2label": {
|
33 |
"0": "Aortic enlargement",
|
34 |
+
"1": "Cardiomegaly",
|
35 |
+
"2": "Nodule/Mass",
|
36 |
+
"3": "Pleural thickening",
|
37 |
+
"4": "Pulmonary fibrosis",
|
38 |
+
"5": "No finding"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
},
|
40 |
"init_std": 0.02,
|
41 |
"init_xavier_std": 1.0,
|
42 |
"is_encoder_decoder": true,
|
43 |
"label2id": {
|
44 |
"Aortic enlargement": 0,
|
45 |
+
"Cardiomegaly": 1,
|
46 |
+
"No finding": 5,
|
47 |
+
"Nodule/Mass": 2,
|
48 |
+
"Pleural thickening": 3,
|
49 |
+
"Pulmonary fibrosis": 4
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
50 |
},
|
51 |
"mask_loss_coefficient": 1,
|
52 |
"max_position_embeddings": 1024,
|
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:9308365ccef2aea0b004456a05445ad134ce42a4247ba08acfe4c3296a188170
|
3 |
+
size 166499964
|
runs/Apr01_18-00-54_6ae021626422/events.out.tfevents.1711994494.6ae021626422.5756.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2468f151f38b33f305f92d7ee888b40ca9bac837a7ca568d646f904c84c44d98
|
3 |
+
size 6078
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 4920
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8f61c562c978a6d78a9372aa371b19c42578a3a936b29e3af5019b7e5a24ffa7
|
3 |
size 4920
|