justinpinkney / stable-diffusion

MIT License
1.45k stars 266 forks source link

Set max steps or epochs #56

Open Conaire opened 1 year ago

Conaire commented 1 year ago

Hey,

How do we set the max steps or epochs?

Thanks

justinpinkney commented 1 year ago

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!)