Low precision neural networks

0 0
  • 0 Collaborators

This project aims to implement an optimized framework for XNOR-nets ...learn more

HPC, Artificial Intelligence

Groups
Student Developers for AI

Overview / Usage

https://github.com/akhauriyash/XNOR-convolution

This project hopes to implement an optimized model of a quantized neural network.

TO DO:

  • [x] Add support for variable input sizes
  • [x] Add support for 3D convolution
  • [x] Parallelize per convolution
  • [ ] Maximize shared memory usage - balance channel parallelization
  • [ ] Create a full precision verification kernel
  • [ ] Add full support for custom kernel sizes
  • [ ] Build a parser to take in shape arguments
  • [ ] Add in proof of concept for densely connected networks
Comments (0)