training can be performed on the CPU as well#7

psteinb wants to merge 1 commit intovislearn:mastervislearn/conditional_INNs:masterfrom

psteinb:flexible-data-downloadpsteinb/conditional_invertible_neural_networks:flexible-data-downloadCopy head branch name to clipboard