CumConcatLayer, small fix

rwth-i6 · Feb 15, 2022 · 916d70d · 916d70d
1 parent 5d3d0cb
commit 916d70d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/returnn/tf/layers/rec.py b/returnn/tf/layers/rec.py
@@ -9287,7 +9287,7 @@ class CumConcatLayer(_ConcatInputLayer):
   layer_class = "cum_concat"
   recurrent = True  # order matters
 
-  def __init__(self, out_spatial_dim, axis=NotSpecified, **kwargs):
+  def __init__(self, out_spatial_dim, axis=None, **kwargs):
     """
     :param Dim out_spatial_dim:
     :param Dim|None axis: to operate over. only single_step_dim supported currently, assumes to be inside rec layer