Hi @aatiibutt
Its not good to see train loss increase. You can add Dropout() and batchnorm1d to your linear layers.
Make sure to give shuffle=true in the train dataloaders. Try these changes.
Its always a good practice to have a validation set which evaluates after each epoch.
Let me know if this worked.