lucidrains/imagen-pytorch

Improvements for beginners

Open

#307 opened on Jan 26, 2023

View on GitHub
 (8 comments) (2 reactions) (0 assignees)Python (690 forks)batch import
good first issuehelp wanted

Repository metrics

Stars
 (7,324 stars)
PR merge metrics
 (No merged PRs in 30d)

Description

  • Add Documentation to set up a training loop
  • Support for webdataset
  • slurm (as people wanted it in the LAION discord)
  • Docs for advanced parameters
  • Requirements (for small scale and large scale training)
  • preencoding ( ability to use seperate Cuda device )

Contributor guide