baal-org / baal

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

Training Args for ModelWrapper #289

Closed Dref360 closed 1 month ago

Dref360 commented 2 months ago

In a further effort to align APIs, ModelWrapper should have a TrainingArgs-like argument. Would simplify a lot of things across the codebase.