soumik12345 / wandb-addons

Weights & Biases Addons is a repository consisting of additional unitilities and community contributions for supercharging your Weights & Biases workflows.
https://geekyrakshit.dev/wandb-addons/
23 stars 7 forks source link

[feature] Add W&B Eval Callback and W&B GradCAM Callback #27

Closed ayulockin closed 1 year ago

ayulockin commented 1 year ago

I have added two new opinionated Keras callbacks. The dir structure and API design is something we can discuss further but wanted to get it out crudely for the event in context.

I have loosely type hinted the classes. Do let me know if any quick fix is required.

The callbacks were manually tested.

PS: The grad cam callback can be built by subclassing the eval callback. One can also argue to have the eval callback have one argument added to it for explainability bit. But thought of having two separate class (with not much code diff) for now. We can discuss the design more.

cc: @soumik12345