Is fastAI supported - Githubissues

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

Open vahuja4 opened 2 years ago

vahuja4 commented 2 years ago

I am wanting to know if baal supports the fastAI library? If so, please share an example.

Dref360 commented 2 years ago

Hello,

I'm not familiar with fastAI unfortunatly.

But let me know if that works for you:

ActiveLearningDataset works on a pytorch Dataset.
ActiveLearningLoop requires a function that returns the predictions from a Pytorch Dataset.
Heuristics like BALD works on any array numpy or tensors.

We would love to know more about this potential integration.

We already support HugginFace and Pytorch-Lightning if that can help!

Dref360 commented 2 years ago

Anything I can assist with? Happy to schedule a call as needed too.