VGG: fix wrong kwargs in get_config

ctu-geoforall-lab-projects · Sep 22, 2023 · cd75981 · cd75981 · github-actions · Sep 22, 2023
1 parent 94c2b49
commit cd75981
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/architectures.py b/src/architectures.py
@@ -1027,8 +1027,8 @@ def get_config(self):
         """
         config = super(VGG, self).get_config()
 
-        config.update(pooling=self.pooling,
-                      depth=self.depth,
+        config.update(depth=self.depth,
+                      include_top=include_top,
                       return_layers=self.return_layers)
 
         return config