[Tensorflow/LanguageModelling/BERT] Step is incremented incorrectly in LAMB optimizer #391

Related to Model/Framework(s)
Tensorflow/LanguageModelling/BERT LAMB optimizer

Describe the bug
On this line the step is incremented every time the new variable is updated but it is supposed to be the global step
which leads to incorrect bias correction.