Delete modeling_customefficientnetv2.py

modeling_customefficientnetv2.py

@@ -1,27 +0,0 @@
-from transformers import PretrainedModel
-from configuration_customefficientnetv2 import CustomEfficientNetV2Config
-import torch
-
-class CustomEfficientNetV2(PretrainedModel):
-    config_class = CustomEfficientNetV2Config
-
-    def __init__(self, config):
-        super().__init__(config)
-
-        self.url = config.url
-        file_name = self.url.split('/')[-1]
-        self.model = torch.load(file_name)
-        
-        self.input_size = config.input_size
-        shape = [2] + self.input_size
-        example_inputs = torch.randn(shape)
-        example_inputs = (example_inputs - example_inputs.min()) / (example_inputs.max() - example_inputs.min())
-
-        self.num_classes = config.num_classes
-        if self.num_classes != 1000:
-            self.model.classifier = torch.nn.Linear(in_features=1984, out_features=self.num_classes, bias=True)
-        
-        traced_model = torch.jit.trace(self.model, example_inputs)
-        traced_model.save(file_name)
-
-        self.model = torch.jit.load(file_name)