Skip to content

Commit 1f356f0

Browse files
committed
RecLayer, expose time_dim_tag
1 parent 9648b4c commit 1f356f0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

returnn/tf/layers/rec.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def __init__(self,
147147
self._initial_state_deps = [layer for layer in nest.flatten(initial_state) if isinstance(layer, LayerBase)]
148148
self._input_projection = input_projection
149149
self._max_seq_len = max_seq_len
150-
self._output_time_dim_tag = _time_dim_tag
150+
self.time_dim_tag = _time_dim_tag
151151
self.include_eos = include_eos
152152
if optimize_move_layers_out is None:
153153
optimize_move_layers_out = self.network.get_config().bool("optimize_move_layers_out", True)

0 commit comments

Comments
 (0)