Learning is not happening, exponential growth of theta variables