iammbrn
/

alsatix_image_control_v2

Model card Files Files and versions Community

iammbrn commited on Jan 4

Commit

c104af3

·

verified ·

1 Parent(s): 9a9f89d

Update model_config.json

Files changed (1) hide show

model_config.json +2 -5

model_config.json CHANGED Viewed

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

 {
+    "model_type": "resnet",
     "num_labels": 5,
     "id2label": {
         "0": "alkol",
     "input_size": 224,
     "num_classes": 5,
     "pretrained": true,
+    "architecture": "ResNet50"
 }