Selecting Data for Batches

[golnooshAbd]

LSTM_encoder_decoder/code/lstm_encoder_decoder.py

Line 160 in 79a6ece

input_batch = input_tensor[:, b: b + batch_size, :]

LSTM_encoder_decoder/code/lstm_encoder_decoder.py

Line 161 in 79a6ece

target_batch = target_tensor[:, b: b + batch_size, :]

Basically these lines are wrong and should be changed like below to iterate on all samples, although, it still doesn't consider the last samples which are not fit into (less than) one batch.

input_batch = input_tensor[:, b * batch_size: (b + 1) * batch_size, :]
target_batch = target_tensor[:, b * batch_size: (b + 1) * batch_size, :]