Training in progress, step 100

config.json

@@ -39,39 +39,15 @@
   "giou_cost": 2,
   "giou_loss_coefficient": 2,
   "id2label": {
-    "0": "Aortic enlargement",
-    "1": "Atelectasis",
-    "2": "Calcification",
-    "3": "Cardiomegaly",
-    "4": "Consolidation",
-    "5": "ILD",
-    "6": "Infiltration",
-    "7": "Lung Opacity",
-    "8": "Nodule/Mass",
-    "9": "Other lesion",
-    "10": "Pleural effusion",
-    "11": "Pleural thickening",
-    "12": "Pneumothorax",
-    "13": "Pulmonary fibrosis"
+    "0": "no_tb",
+    "1": "tb"
   },
   "init_std": 0.02,
   "init_xavier_std": 1.0,
   "is_encoder_decoder": true,
   "label2id": {
-    "Aortic enlargement": 0,
-    "Atelectasis": 1,
-    "Calcification": 2,
-    "Cardiomegaly": 3,
-    "Consolidation": 4,
-    "ILD": 5,
-    "Infiltration": 6,
-    "Lung Opacity": 7,
-    "Nodule/Mass": 8,
-    "Other lesion": 9,
-    "Pleural effusion": 10,
-    "Pleural thickening": 11,
-    "Pneumothorax": 12,
-    "Pulmonary fibrosis": 13
+    "no_tb": 0,
+    "tb": 1
   },
   "mask_loss_coefficient": 1,
   "max_position_embeddings": 1024,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7cd6814ceeadb45ea6fe3926d2b3d76fe34934e53adea163acb6076582325082
-size 166508196
+oid sha256:4a921cd5dbed003f068b03774f631d8361e23ce86dbeb2f199ef116d86d72b9d
+size 166495852

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c159a4f4a8b69ed9088a43843124886064a560618aa6eb7aa890eeba40d811f7
+oid sha256:9ffb614cdf305cab56e3e8c32796642c1de47300cd79bef105e5999eb25599f2
 size 5368