Tensorflow 2.3: Neural netowrk Regression problem - Cost not changing

Question

I am working on a Neural Network which can predict state of charge of Li-ion cell based on usage condition. Here is the pictorial representation.

I have modified code