Pytorch Toolbox

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 Linear layer into Conv2D
• Torch to pytorch: convert torch model into pytorch model y
• Visualize model: generate model graph