lucidrains/imagen-pytorch

Improvements for beginners

Open

#307 ouverte le 26 janv. 2023

Voir sur GitHub
 (8 commentaires) (2 réactions) (0 assignés)Python (690 forks)batch import
good first issuehelp wanted

Métriques du dépôt

Stars
 (7 324 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

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 )

Guide contributeur