Upload config.json with huggingface_hub

config.json

@@ -0,0 +1,26 @@
+{
+  "model_type": "grain_classifier",
+  "architecture": "ResNet-50",
+  "task": "multi_task_classification",
+  "tasks": [
+    "count_prediction",
+    "good_grain_count",
+    "bad_grain_count",
+    "quality_classification"
+  ],
+  "input_size": [
+    3,
+    256,
+    256
+  ],
+  "num_classes": 2,
+  "training_info": {
+    "best_epoch": 29,
+    "train_loss": 19.699030657283597,
+    "val_loss": 17.164573192596436,
+    "optimizer": "AdamW",
+    "scheduler": "OneCycleLR",
+    "gpu": "Tesla T4",
+    "mixed_precision": true
+  }
+}