Training in progress, step 500

config.json

@@ -15,97 +15,13 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 384,
   "id2label": {
-    "0": "N/A",
-    "1": "person",
-    "2": "bicycle",
-    "3": "car",
-    "4": "motorcycle",
-    "5": "airplane",
-    "6": "bus",
-    "7": "train",
-    "8": "truck",
-    "9": "boat",
-    "10": "traffic light",
-    "11": "fire hydrant",
-    "12": "N/A",
-    "13": "stop sign",
-    "14": "parking meter",
-    "15": "bench",
-    "16": "bird",
-    "17": "cat",
-    "18": "dog",
-    "19": "horse",
-    "20": "sheep",
-    "21": "cow",
-    "22": "elephant",
-    "23": "bear",
-    "24": "zebra",
-    "25": "giraffe",
-    "26": "N/A",
-    "27": "backpack",
-    "28": "umbrella",
-    "29": "N/A",
-    "30": "N/A",
-    "31": "handbag",
-    "32": "tie",
-    "33": "suitcase",
-    "34": "frisbee",
-    "35": "skis",
-    "36": "snowboard",
-    "37": "sports ball",
-    "38": "kite",
-    "39": "baseball bat",
-    "40": "baseball glove",
-    "41": "skateboard",
-    "42": "surfboard",
-    "43": "tennis racket",
-    "44": "bottle",
-    "45": "N/A",
-    "46": "wine glass",
-    "47": "cup",
-    "48": "fork",
-    "49": "knife",
-    "50": "spoon",
-    "51": "bowl",
-    "52": "banana",
-    "53": "apple",
-    "54": "sandwich",
-    "55": "orange",
-    "56": "broccoli",
-    "57": "carrot",
-    "58": "hot dog",
-    "59": "pizza",
-    "60": "donut",
-    "61": "cake",
-    "62": "chair",
-    "63": "couch",
-    "64": "potted plant",
-    "65": "bed",
-    "66": "N/A",
-    "67": "dining table",
-    "68": "N/A",
-    "69": "N/A",
-    "70": "toilet",
-    "71": "N/A",
-    "72": "tv",
-    "73": "laptop",
-    "74": "mouse",
-    "75": "remote",
-    "76": "keyboard",
-    "77": "cell phone",
-    "78": "microwave",
-    "79": "oven",
-    "80": "toaster",
-    "81": "sink",
-    "82": "refrigerator",
-    "83": "N/A",
-    "84": "book",
-    "85": "clock",
-    "86": "vase",
-    "87": "scissors",
-    "88": "teddy bear",
-    "89": "hair drier",
-    "90": "toothbrush"
+    "0": "metals_and_plastic",
+    "1": "other",
+    "2": "non_recyclable",
+    "3": "glass",
+    "4": "paper",
+    "5": "bio",
+    "6": "unknown"
   },
   "image_size": [
     512,
@@ -114,87 +30,13 @@
   "initializer_range": 0.02,
   "intermediate_size": 1536,
   "label2id": {
-    "N/A": 83,
-    "airplane": 5,
-    "apple": 53,
-    "backpack": 27,
-    "banana": 52,
-    "baseball bat": 39,
-    "baseball glove": 40,
-    "bear": 23,
-    "bed": 65,
-    "bench": 15,
-    "bicycle": 2,
-    "bird": 16,
-    "boat": 9,
-    "book": 84,
-    "bottle": 44,
-    "bowl": 51,
-    "broccoli": 56,
-    "bus": 6,
-    "cake": 61,
-    "car": 3,
-    "carrot": 57,
-    "cat": 17,
-    "cell phone": 77,
-    "chair": 62,
-    "clock": 85,
-    "couch": 63,
-    "cow": 21,
-    "cup": 47,
-    "dining table": 67,
-    "dog": 18,
-    "donut": 60,
-    "elephant": 22,
-    "fire hydrant": 11,
-    "fork": 48,
-    "frisbee": 34,
-    "giraffe": 25,
-    "hair drier": 89,
-    "handbag": 31,
-    "horse": 19,
-    "hot dog": 58,
-    "keyboard": 76,
-    "kite": 38,
-    "knife": 49,
-    "laptop": 73,
-    "microwave": 78,
-    "motorcycle": 4,
-    "mouse": 74,
-    "orange": 55,
-    "oven": 79,
-    "parking meter": 14,
-    "person": 1,
-    "pizza": 59,
-    "potted plant": 64,
-    "refrigerator": 82,
-    "remote": 75,
-    "sandwich": 54,
-    "scissors": 87,
-    "sheep": 20,
-    "sink": 81,
-    "skateboard": 41,
-    "skis": 35,
-    "snowboard": 36,
-    "spoon": 50,
-    "sports ball": 37,
-    "stop sign": 13,
-    "suitcase": 33,
-    "surfboard": 42,
-    "teddy bear": 88,
-    "tennis racket": 43,
-    "tie": 32,
-    "toaster": 80,
-    "toilet": 70,
-    "toothbrush": 90,
-    "traffic light": 10,
-    "train": 7,
-    "truck": 8,
-    "tv": 72,
-    "umbrella": 28,
-    "vase": 86,
-    "wine glass": 46,
-    "zebra": 24
+    "bio": 5,
+    "glass": 3,
+    "metals_and_plastic": 0,
+    "non_recyclable": 2,
+    "other": 1,
+    "paper": 4,
+    "unknown": 6
   },
   "layer_norm_eps": 1e-12,
   "model_type": "yolos",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8ad3766f492aae940d3721d749cf5f7066eb27f6c126ba081e9c8f40ac462b90
-size 122810973
+oid sha256:5740cd83712d9a13af8b49dbd9a40d639ca21c0baae5a3ae913890ebf77e7e8e
+size 122681629

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:906f79bc8091a185e17cc66ee0f97cfd8e561c301f479e8e7342ba5717b2c1b7
+oid sha256:fce898a6ff528685a2e98b25230fc415e5e9a7025152019108b775f4730b7511
 size 4027