Training based on iterations

[mshooter]

Hi,
Could anyone advice me how to set up PyTorch lightning trainer to learn based on iterations instead of epochs?

Thank you!

[ananthsub]

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

[mshooter]

Ok great thanks!