fix 'VGG' object has no attribute 'vgg_layers'

ctu-geoforall-lab-projects · Aug 27, 2023 · cd0afd6 · cd0afd6
1 parent 7519687
commit cd0afd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/architectures.py b/src/architectures.py
@@ -977,7 +977,7 @@ def __init__(self, *args, depth=16, include_top=True,
 
         super(VGG, self).__init__(*args, **kwargs)
 
-        self.instantiate_layers()
+        self.vgg_layers = self.instantiate_layers()
 
 
     def call(self, inputs, training=None, mask=None):