infinite while loop in TrainLoop.run_loop

[tahaceritli]

Hi,

Thanks for sharing this repo!

I am trying to train the models from scratch. It seems that TrainLoop(...).run_loop() function has a while loop that is running on infinitely, as self.lr_anneal_steps is set 0. I'm referring to the code snippet given below: while ( not self.lr_anneal_steps or self.step + self.resume_step < self.lr_anneal_steps ):

I suppose that it is safe to define a num_iterations parameter and use it in this loop? If so, I suppose this parameter is the one reported in Table 11-13 of the arXiv paper.

I would appreciate if you could clarify this.

Best, Taha