Currently the toy task is to sum numbers. While it runs fast, it is not ideal for visualizing results, testing confusion matrices, displaying images, etc.
We are now replacing this with a more "visual" task, e.g. fashionMNIST
They are a few changes I believe we should make (dataclass, factories, names), but they should be kept for future PR. This one is large enough, and more focussed discussion is needed.
Currently the toy task is to sum numbers. While it runs fast, it is not ideal for visualizing results, testing confusion matrices, displaying images, etc.
We are now replacing this with a more "visual" task, e.g. fashionMNIST