baal-org / baal

Bayesian active learning library for research and industrial usecases.
https://baal.readthedocs.io
Apache License 2.0
862 stars 86 forks source link

Add ability to unpatch a Module #189

Closed Dref360 closed 2 years ago

Dref360 commented 2 years ago

We need a per-technique way of resetting a Module.

For bayesian.dropout we need to reset the Dropout Layer For bayesian.dropconnect we need to make the layers deterministic again.