Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/master'
Browse files Browse the repository at this point in the history
  • Loading branch information
Albert Aparicio committed Mar 7, 2017
2 parents 6b8d79c + 25eeb4b commit dd9e5ad
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -243,3 +243,4 @@ nytprof.out
*.o
*.bs
/_eumm/
/unused_stuff/
8 changes: 4 additions & 4 deletions tfglib/pretrain_data_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -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],
Expand Down

0 comments on commit dd9e5ad

Please sign in to comment.