This repository contains several tools useful for pytorch users.
- Cuda to cpu: get rid of dataparallel while loading cuda model into cpu
- Pytorch to thnets: convert pytorch model into thnets representation, tested to work with pytorch version 0.2.0+08b4770
- Spatial model: convert
Linearlayer intoConv2D - Torch to pytorch: convert torch model into pytorch model y
- Visualize model: generate model graph