Thank you for pointing out the overfitting issue. I wasn’t sure how to tackle it. The change suggested by @patrickwilliams3 definitely helped in converging the val_loss better. However, my model is still overfitting so I’ll try out the transformations suggested by you.