Open Conaire opened 1 year ago
Hey,
How do we set the max steps or epochs?
Thanks
there is a pytorch lightning option called for the trainer. You can set this via yaml or command line, trainer.max_steps=1000 (check the pytorch lightning docs for what the actual argument is called though!)
trainer.max_steps=1000
Hey,
How do we set the max steps or epochs?
Thanks