Skip to content

Commit 7a28a2b

Browse files
committed
remove return_last_state warning for RNN
1 parent 15d48b5 commit 7a28a2b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tensorlayer/layers/recurrent.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -247,11 +247,10 @@ def forward(self, inputs, sequence_length=None, initial_state=None, **kwargs):
247247
sequence_length = [i - 1 for i in sequence_length]
248248

249249
# set warning
250-
if (not self.return_last_state or not self.return_last_output) and sequence_length is not None:
250+
if (not self.return_last_output) and sequence_length is not None:
251251
warnings.warn(
252252
'return_last_output is set as %s ' % self.return_last_output +
253-
'and return_last_state is set as %s. ' % self.return_last_state +
254-
'When sequence_length is provided, both are recommended to set as True. ' +
253+
'When sequence_length is provided, it is recommended to set as True. ' +
255254
'Otherwise, padding will be considered while RNN is forwarding.'
256255
)
257256

0 commit comments

Comments
 (0)