When using "transformers.WarmUp" with tensorflow 2.0.0, warming up restart in each epoch!
When using "transformers.WarmUp" with tensorflow 2.0.0, warming up restart in each epoch!
That is because, in keras "Callbacks.on_batch_begin(self, batch, logs)", batch start from zero in each epoch when using "fit" method.
Hello!

Can you give a code that better explain your issue and a way to reproduce it please? Thanks :) <|||||>Sorry! I check again and it works well!
Close it!
