lightning - Serbipunk/notes GitHub Wiki

relationship with python_lightning

https://github.com/Lightning-AI/pytorch-lightning/discussions/16688

debuging

pdb.set_trace()

https://lightning.ai/docs/pytorch/stable/debug/debugging_advanced.html

pix2pix example

https://www.aryan.no/post/pix2pix/pix2pix/

multiple optimizers

https://lightning.ai/docs/pytorch/stable/model/manual_optimization.html#use-multiple-optimizers-like-gans

set end of epoch

torchvision.utils.save_image(grid_image, fp=f'{trainer.log_dir}/epoch-{trainer.current_epoch:04}.png')

plot networks

txt

plot networks

dot

``