Update convnext.py

neuronets · Mar 20, 2024 · dffbaa4 · dffbaa4
1 parent ead3620
commit dffbaa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nobrainer/models/convnext.py b/nobrainer/models/convnext.py
@@ -32,7 +32,7 @@ def call(self, x, training=None):
         return drop_path(x, self.drop_prob, training)
 
 
-class Block(Layer):
+class Block(layers.Layer):
     """ConvNeXt Block. There are two equivalent implementations:
     (1) DwConv -> LayerNorm (channels_first)
     -> 1x1x1 Conv -> GELU -> 1x1x1 Conv; all in (N, C, H, W, D)