search

CompVis / latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models
MIT License
11.87k stars 1.53k forks source link

About training epoch #194

Open pokameng opened 1 year ago

pokameng commented 1 year ago

I can't find the epoch set for the training in .yaml file (e.g. lsun_bedroom-ldm-vq-4.yaml). How do we now the epoch of training and change it? How many epochs will the programe end? @rromb @asanakoy @pesser @crowsonkb @theRealSuperMario @to3i

kapsner commented 1 year ago

You would need to set max_epochs in the lightning configs, e.g.

lightning:
  trainer:
    gpus: 0,
    log_every_n_steps: 15
    max_epochs: 400

This defaults to 1000: https://pytorch-lightning.readthedocs.io/en/stable/common/trainer.html#max-epochs