IntelLabs / bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How bayesian network be used in quantization?

LeopoldACC opened this issue · comments

@peteriz @jpablomch @ranganathkrishnan we can get the $\mu$ and $\sigma$ of each weight ,so how it can be used to quantization which mapping the $w_{float32 }$ into $w_{int8}$

@LeopoldACC Quantization of variational parameters in the Bayesian layers is not supported currently.

@LeopoldACC Quantization of Bayesian deep neural network models are supported in the latest release v0.4.0. Please check out!