SebChw / Actually-Robust-Training

Actually Robust Training - Tool Inspired by Andrej Karpathy "Recipe for training neural networks". It allows you to decompose your Deep Learning pipeline into modular and insightful "Steps". Additionally it has many features for testing and debugging neural nets.
MIT License
44 stars 0 forks source link

Write sqeeze step -> ensemble step #147

Open kordc opened 1 year ago

kordc commented 1 year ago

I'd focus there on the 2 steps he tells us about:

So maybe instead of the squeeze, we should enable ensemble DL training?

kordc commented 1 year ago

We agreed on making just ensemble step

kordc commented 1 year ago

Deep learning ensembles' training is a huge topic. I found a nice package Ensembles-PyTorch. They provide a wide selection of already implemented ensemble types. I'll play around with this library using our ArtModule

SebChw commented 1 year ago

Nice, if we won't need to write it by ourselves it would be great.

kordc commented 1 year ago

It supports only classification and regression. Even multi-label classification is not supported.

I think doing this step a task-agnostic is simply not possible. I see three options:

trebacz626 commented 1 year ago

I think weighted mean might be good enough

trebacz626 commented 1 year ago

user can specify weights

mmaecki commented 1 year ago

Same as @trebacz626 suggests. Any more sophisticated ensembles can be done by the user.

SebChw commented 1 year ago

Actually, most tasks are classification or regression. If we can utilize this library quickly/easily lets use it, even though it doesn't support everything. If user wants something more sophisticated they must implement it by themselves.