Mukesh1729:

During training my my training and validation loss doesnt improve much


I would recommend to scale down the use case and try to overfit a small dataset (e.g. just 10 samples) by playing around with some hyperparameters and make sure your model is able to learn them. Once this is done you could scale up the use case again.
PS: you can post code snippets by wrapping them into three backticks ```, which makes debugging easier.