Upload 5 files

config.json

@@ -1,67 +1,5 @@
 {
-  "return_dict": true,
-  "output_hidden_states": false,
-  "output_attentions": false,
-  "torchscript": false,
-  "torch_dtype": "float32",
-  "use_bfloat16": false,
-  "tf_legacy_loss": false,
-  "pruned_heads": {},
-  "tie_word_embeddings": true,
-  "chunk_size_feed_forward": 0,
-  "is_encoder_decoder": false,
-  "is_decoder": false,
-  "cross_attention_hidden_size": null,
-  "add_cross_attention": false,
-  "tie_encoder_decoder": false,
-  "max_length": 20,
-  "min_length": 0,
-  "do_sample": false,
-  "early_stopping": false,
-  "num_beams": 1,
-  "num_beam_groups": 1,
-  "diversity_penalty": 0.0,
-  "temperature": 1.0,
-  "top_k": 50,
-  "top_p": 1.0,
-  "typical_p": 1.0,
-  "repetition_penalty": 1.0,
-  "length_penalty": 1.0,
-  "no_repeat_ngram_size": 0,
-  "encoder_no_repeat_ngram_size": 0,
-  "bad_words_ids": null,
-  "num_return_sequences": 1,
-  "output_scores": false,
-  "return_dict_in_generate": false,
-  "forced_bos_token_id": null,
-  "forced_eos_token_id": null,
-  "remove_invalid_values": false,
-  "exponential_decay_length_penalty": null,
-  "suppress_tokens": null,
-  "begin_suppress_tokens": null,
-  "architectures": [
-    "CustomEfficientNet"
-  ],
-  "finetuning_task": null,
-  "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1"
-  },
-  "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1
-  },
-  "tokenizer_class": null,
-  "prefix": null,
-  "bos_token_id": null,
-  "pad_token_id": null,
-  "eos_token_id": null,
-  "sep_token_id": null,
-  "decoder_start_token_id": null,
-  "task_specific_params": null,
-  "problem_type": null,
-  "_name_or_path": "",
-  "transformers_version": "4.41.2",
+  "model_type": "custom_efficientnet",
   "num_classes": 2,
-  "model_type": "custom_efficientnet"
-}
+  "transformers_version": "4.41.2"
+}

custom_efficientnet.py

@@ -0,0 +1,34 @@
+from transformers import PreTrainedModel, PretrainedConfig
+from torchvision import models
+import torch.nn as nn
+
+class CustomEfficientNetConfig(PretrainedConfig):
+    model_type = "custom_efficientnet"
+    def __init__(self, num_classes=2, **kwargs):
+        super().__init__(**kwargs)
+        self.num_classes = num_classes
+
+class CustomEfficientNetForImageClassification(PreTrainedModel):
+    config_class = CustomEfficientNetConfig
+    base_model_prefix = "efficientnet"
+
+    def __init__(self, config):
+        super().__init__(config)
+        self.num_labels = config.num_classes
+        self.efficientnet = models.efficientnet_b0(num_classes=config.num_classes)
+        
+    def forward(self, pixel_values, labels=None):
+        outputs = self.efficientnet(pixel_values)
+        loss = None
+        if labels is not None:
+            loss_fct = nn.CrossEntropyLoss()
+            loss = loss_fct(outputs, labels)
+        return {"loss": loss, "logits": outputs}
+
+    @classmethod
+    def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
+        config = CustomEfficientNetConfig.from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
+        model = cls(config)
+        state_dict = torch.load(pretrained_model_name_or_path + "/pytorch_model.bin")
+        model.load_state_dict(state_dict)
+        return model

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b3aa5cb431970b4a34fddda4069a0be17f15e648ba9020a4b514957742a237f
+size 256569792

preprocessor_config.json

@@ -1,47 +1,15 @@
 {
-  "_valid_processor_keys": [
-    "images",
-    "do_resize",
-    "size",
-    "resample",
-    "do_center_crop",
-    "crop_size",
-    "do_rescale",
-    "rescale_factor",
-    "rescale_offset",
-    "do_normalize",
-    "image_mean",
-    "image_std",
-    "include_top",
-    "return_tensors",
-    "data_format",
-    "input_data_format"
-  ],
-  "crop_size": {
-    "height": 289,
-    "width": 289
-  },
-  "do_center_crop": false,
-  "do_normalize": true,
-  "do_rescale": true,
   "do_resize": true,
+  "size": 224,
+  "do_normalize": true,
   "image_mean": [
     0.485,
     0.456,
     0.406
   ],
-  "image_processor_type": "EfficientNetImageProcessor",
   "image_std": [
     0.229,
     0.224,
     0.225
-  ],
-  "include_top": true,
-  "resample": 0,
-  "rescale_factor": 0.00392156862745098,
-  "rescale_offset": false,
-  "size": {
-    "height": 600,
-    "width": 600
-  }
-}
+  ]
+}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:146cbd6be07e82ae04954ac25ed6ec4dd0af7acdbc59f35ad50d34afb1b33b3f
+size 16344682