How to use the baal.utils.metrics.Loss 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 / modelwrapper.py View on Github external
        self.add_metric('loss', lambda: Loss())
        self.replicate_in_memory = replicate_in_memory