How to use the baal.bayesian.bayesian_layer.LinearBayesianLayer function in baal

To help you get started, we’ve selected a few baal examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github ElementAI / baal / src / baal / bayesian / bayesian_layer / wrapper.py View on Github external
def reset(m):
            if isinstance(m, LinearBayesianLayer):
                m.reset_parameters()
        self.model.apply(reset)