Add Args to ModelWrapper to simplify common API

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 Args to ModelWrapper to simplify common API #294

Closed Dref360 closed 4 months ago

Dref360 commented 4 months ago

Summary:

Similar to other great libraries (PL or HF) we can simplify how we manage our Args. This is a major breaking change hence why keeping it for Baal 2.0.

Features:

Fixes #289

Checklist:

[ ] Your code is documented (To validate this, add your module to tests/documentation_test.py).
[ ] Your code is tested with unit tests.
[ ] You moved your Issue to the PR state.