diff --git a/.gitignore b/.gitignore index caa2508..ed58e8b 100644 --- a/.gitignore +++ b/.gitignore @@ -243,3 +243,4 @@ nytprof.out *.o *.bs /_eumm/ +/unused_stuff/ diff --git a/tfglib/pretrain_data_params.py b/tfglib/pretrain_data_params.py index bbb4e69..8d4c94c 100644 --- a/tfglib/pretrain_data_params.py +++ b/tfglib/pretrain_data_params.py @@ -363,12 +363,12 @@ def prepare_pretrain_slice( feedback_data = np.roll(trg_res, 1, axis=0) feedback_data[0, :] = 0 - # Flip slice frames and return them + # Return slice frames # print('Sliced ' + basename) yield ( - np.fliplr(src_res[:, 0:44]), - np.fliplr(src_res[:, 44:45]).reshape((-1)), - np.fliplr(src_res[:, 45:46]).reshape((-1)), + src_res[:, 0:44], + src_res[:, 44:45].reshape((-1)), + src_res[:, 45:46].reshape((-1)), feedback_data, trg_res[:, 0:42], trg_res[:, 42:44],