Redo how predict works, fix reversing
This reverts commit a5b885ecd4.
a5b885ecd4
The branches are getting messy! This branch is a clean updated version of the codebase which is actually compatible with 1) new version of pytorch 2) Almond project