softmax bug fix

imdeepmind · Feb 23, 2021 · c278a72 · c278a72
1 parent 6e5af2a
commit c278a72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralpy/layers/activation_functions/softmax.py b/neuralpy/layers/activation_functions/softmax.py
@@ -29,7 +29,7 @@ def __init__(self, dim=None, name=None):
         """
         super().__init__(_Softmax, "Softmax", layer_name=name)
 
-        if not isinstance(dim, int):
+        if dim is not None and not isinstance(dim, int):
             raise ValueError("Please provide a valid dim")
 
         self.__dim = dim