Low precision neural networks
Yash Akhauri
Unknown
- 0 Collaborators
This project aims to implement an optimized framework for XNOR-nets ...learn more
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