search

KarenUllrich / Tutorial_BayesianCompressionForDL

A tutorial on "Bayesian Compression for Deep Learning" published at NIPS (2017).
MIT License
203 stars 48 forks source link

How to use bayesianConv2d? #6

Closed RichardMrLu closed 5 years ago

RichardMrLu commented 6 years ago

Can you show a example of bayesianLayers.Conv2dGroupNJ()? I want to use this conv on my network but there is some error. Thanks for your help.

KarenUllrich commented 6 years ago

HI RichardMrLu,

I'll refer you to issue #2 . I posted a complete example in the end.

Cheers, Karen