pytorch

723fcfaf - implement post-rewrite model testing