Training based on iterations
Hi,
Could anyone advice me how to set up PyTorch lightning trainer to learn based on iterations instead of epochs?
Thank you!
Hi @mshooter , you can use the min_steps and max_steps arguments on the Trainer to do training based on iterations instead of epochs. https://pytorch-lightning.readthedocs.io/en/latest/common/trainer.html#max-steps