Mis match error in pytorch module

[Before posting a query, check the FAQs - it might already be answered!](Google Colab)

Your linear layer is raising the shape mismatch so make sure the input activation to this layer matches the in_features.
PS: you can post code snippets by wrapping them into three backticks ```, which makes debugging easier.