@googlebot . thanks for replying .

i will  implement a scheduler .
i tried printing my loss while the gradient descent is running , it seems to initially fall down and then it stays constant at not so low value without any change
my X is 0 mean unit variance (unit normal distribution) so i think scaling shouldnt be an issue . please let me know if i understood this wrong . what do you mean by epoch 0 prediction is too far off?