Omniglot dataset using MLP model got shape error during training.

[jocelynbaduria]

🐛 Describe the bug A clear and concise description of what the bug is.

I tried avalanche using sample code from your GitHub and tutorials. I tried to implement Omniglot dataset using MLP model. I am getting this error.

🐜 To Reproduce Steps / minimal snipped of code to reproduce the issue.

🐝 Expected behavior A clear and concise description of what you expected to happen.

🐞 Screenshots If applicable, add screenshots to help explain your problem.

🦋 Additional context Add any other context about the problem here like your python setup.

[jocelynbaduria]

Attach is the colab file colab

[AndreaCossu]

We need authorization to access the colab. Please, check that the input size of your model matches the Omniglot input size (e.g., if you are using the SimpleMLP you may want to change the input size when creating it).

[jocelynbaduria]

We need authorization to access the colab. Please, check that the input size of your model matches the Omniglot input size (e.g., if you are using the SimpleMLP you may want to change the input size when creating it).

Hi please tried it again if you can access now. I did try to change the input size to 28 x 28 but not sure about the batch size to have a total input of size 352800.