rethinking-bnn-optimization
rethinking-bnn-optimization copied to clipboard
PyTorch Implementation
Thanks for really good paper.
Do you plan to implement it as a PyTorch version?
Thanks, good to hear you liked the paper!
We won't publish a PyTorch implementation. However, we have recently added Bop to our BNN training library larq - you can find the documentation here. Larq is built on tf.keras so hopefully it will be easy to use.
Please let us know if you have any other issues.
You can also checkout https://github.com/plumerai/larq/issues/95 to follow our reasoning on why we opted for TensorFlow instead of a different framework. In any case we're happy to answer any questions you might have related to the paper or our library Larq.