pytorch based framework for Convolutional Neural Network - View it on GitHub