fix train_test_split quota

Rayhane-mamah · Aug 4, 2018 · c205b45 · c205b45
1 parent 1fbdad0
commit c205b45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tacotron/feeder.py b/tacotron/feeder.py
@@ -44,7 +44,7 @@ def __init__(self, coordinator, metadata_filename, hparams):
 			test_size=test_size, random_state=hparams.tacotron_data_random_state)
 
 		#Make sure test_indices is a multiple of batch_size else round up
-		len_test_indices = self._round_up(len(test_indices), hparams.tacotron_batch_size)
+		len_test_indices = self._round_up(len(test_indices), hparams.tacotron_batch_size) - hparams.tacotron_batch_size
 		extra_test = test_indices[len_test_indices:]
 		test_indices = test_indices[:len_test_indices]
 		train_indices = np.concatenate([train_indices, extra_test])