deJQK / FracBits

Neural Network Quantization With Fractional Bit-widths
11 stars 5 forks source link

Need help with obtaining the bit width assigned to each layer. #3

Closed Ahmad-Jarrar closed 1 year ago

Ahmad-Jarrar commented 1 year ago

I am trying to run your code and will be using your work in an academic project. I ran this experiment but I couldn't see any way provided to get which layers were assigned what bitwidth.

How can I get the bitwidths for each layer in the network, similar to the data used to create figure 2 in the paper (attached below)?

image
deJQK commented 1 year ago

lamda_w and lamda_a from each layer rounded and plotted against the layer index.

Ahmad-Jarrar commented 1 year ago

Thanks