minor updates

app.py

@@ -42,7 +42,9 @@ effnet_hparams = {47: {"num_classes": 2,
                     "dropout":  0.039061686292663655,
                     "width_mult": 0.7540060155156922,
                     "depth_mult": 0.9378692812212488,
-                    "size": "v2_s"},    
+                    "size": "v2_s",
+                    "model_type": "efficientnet_47_planet_detection"
+                    },    
                 61: {
                     "num_classes": 2,
                     "gamma": 0.032606396652426956,
@@ -54,7 +56,9 @@ effnet_hparams = {47: {"num_classes": 2,
                     "dropout": 0.027804120950575217,
                     "width_mult": 1.060782511229692,
                     "depth_mult": 0.7752918857163054,
-                    "size": "v2_s"},
+                    "size": "v2_s",
+                    "model_type": "efficientnet_61_planet_detection"
+                    },
                 75: {
                     "num_classes": 2,
                     "gamma": 0.029768470449465057,
@@ -66,7 +70,8 @@ effnet_hparams = {47: {"num_classes": 2,
                     "dropout":  0.03351826828687193,
                     "width_mult": 1.144132674734038,
                     "depth_mult": 1.2267023928285563,
-                    "size": "v2_s"
+                    "size": "v2_s",
+                    "model_type": "efficientnet_75_planet_detection"
                 }
 }
 # effnet_config = SimpleNamespace(**effnet_hparams)
@@ -245,9 +250,10 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
                                     width_mult=hparams.width_mult,
                                     depth_mult=hparams.depth_mult,
         )
-        EfficientNetConfig.model_type = "efficientnet_%s_planet_detection" % (hparams.num_channels)
+        # EfficientNetConfig.model_type = "efficientnet_%s_planet_detection" % (hparams.num_channels)
+        EfficientNetConfig.model_type = hparams.model_type
     
-    # config.save_pretrained(save_directory=model_loading_name)
+    config.save_pretrained(save_directory=model_loading_name)
     # config = EfficientNetConfig.from_pretrained(model_loading_name)
     
     # model = EfficientNetPreTrained.from_pretrained(model_loading_name)

model_utils/efficientnet_config.py

@@ -242,7 +242,8 @@ class FusedMBConv(nn.Module):
 
 class EfficientNetConfig(PretrainedConfig):
     
-    model_type = "efficientnet"
+    # model_type = "efficientnet"
+    model_type = "efficientnet_61_planet_detection"
     
     def __init__(
         self,