feat: Update main with the latest model

README.md

@@ -0,0 +1,16 @@
+---
+datasets:
+- garythuang/trashnet
+metrics:
+- accuracy
+pipeline_tag: image-classification
+---
+
+# Model Version 1.1
+Created on: 2024-12-20 19:53:42
+
+## Metrics
+- *train_loss*: 0.7815648026353731
+- *train_accuracy*: 72.17412812268118
+- *val_loss*: 0.9300829283893108
+- *val_accuracy*: 67.06231454005935

config.json

@@ -10,7 +10,8 @@
         "RandomRotation(15)",
         "ColorJitter(brightness=0.2, contrast=0.2)",
         "Resize(224x224)",
-        "ToTensor"
+        "ToTensor",
+        "Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])"
     ],
     "optimizer": "Adam",
     "learning_rate": 0.001,

metrics.json

@@ -1,6 +1,6 @@
 {
-    "train_loss": 1.063828223802912,
-    "train_accuracy": 59.4607964382884,
-    "val_loss": 1.1855051144957542,
-    "val_accuracy": 55.78635014836795
+    "train_loss": 0.7815648026353731,
+    "train_accuracy": 72.17412812268118,
+    "val_loss": 0.9300829283893108,
+    "val_accuracy": 67.06231454005935
 }

trashnet.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:90e043cf1574db8d46670f8026c23423dbb90994c7c865cf4caa1058becefe3d
+size 102844888