fix

config/server.das-its-ucm-efficientnet.yaml → config/server.das-its-ucm-vgg.yaml

 general:
-    output_path: "/media/storagecube/gencer/semanticTripletSampling/dumps/das-its-ucm-efficientnet"
+    output_path: "/media/storagecube/gencer/semanticTripletSampling/dumps/das-its-ucm-vgg"
     version: "1"
     clear_old_output: false
 
@@ -9,7 +9,7 @@ general:
     archive_data_size: 420
 
     model_args:
-        model_arch: "efficientnet"
+        model_arch: "vgg"
         feature_size: 256
 
 train:

source/model_ucm.py

@@ -46,15 +46,10 @@ class PredefinedModel(BaseModel):
         return self.final_layer(x)
 
 
-class EfficientNetB0(PredefinedModel):
+class VGG16(PredefinedModel):
     def __init__(self, **kwargs):
-        super(EfficientNetB0, self).__init__(
-            tf.keras.applications.EfficientNet(
-      1.0,
-      1.0,
-      224,
-      0.2,
-      model_name='efficientnetb0'), **kwargs)
+        super(VGG16, self).__init__(
+            tf.keras.applications.VGG16, **kwargs)
 
 class ResNet50v2(PredefinedModel):
     def __init__(self, **kwargs):
@@ -126,8 +121,8 @@ class TripletModel(BaseModel):
             self.internal_model = DenseNet()
         elif model_arch == 'scnn':
             self.internal_model = SCNN()
-        elif model_arch == 'efficientnet':
-            self.internal_model = EfficientNetB0()
+        elif model_arch == 'vgg':
+            self.internal_model = VGG16()
         else:
             raise ValueError(f'Invalid model architecture "{model_arch}".')