Replace toy task with fashion MNIST

mila-iqia / cookiecutter-pyml

MIT License

19 stars 8 forks source link

Replace toy task with fashion MNIST #77

Closed jerpint closed 1 year ago

jerpint commented 1 year ago

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

marondeau-mila commented 1 year ago

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.