lucidrains/imagen-pytorch

Improvements for beginners

Open

#307 aberto em 26 de jan. de 2023

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

Métricas do repositório

Stars
 (7.324 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 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 )

Guia do colaborador