iammbrn
/

alsatix_image_control_v2

Model card Files Files and versions Community

iammbrn commited on Jan 4

Commit

9a9f89d

·

verified ·

1 Parent(s): 047d444

Update model_config.json

Files changed (1) hide show

model_config.json +5 -2

model_config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-    "model_type": "timm_backbone",
     "num_labels": 5,
     "id2label": {
         "0": "alkol",
@@ -18,5 +18,8 @@
     "input_size": 224,
     "num_classes": 5,
     "pretrained": true,
-    "architecture": "ResNet50"
 }

 {
+    "model_type": "resnet50",
     "num_labels": 5,
     "id2label": {
         "0": "alkol",
     "input_size": 224,
     "num_classes": 5,
     "pretrained": true,
+    "architecture": "ResNet50",
+    "optimizer": "Adam",
+    "learning_rate": 0.0001,
+    "epochs": 3
 }