AIForce/train-dalle-together @ 19e3d2d06004a2b88ca1f3daf41d91498baaaa08

Няма описание

Aleksandr Borzunov 19e3d2d060 Make aux peers fetch checkpoints every 2 steps by default		преди 3 години
lib	64dee420da Upgrade to using hivemind.optim.experimental	преди 3 години
.gitignore	72fc0bcdb7 Initial commit (ru-max branch without private code)	преди 3 години
README.md	d4140807f4 Update readme	преди 3 години
arguments.py	19e3d2d060 Make aux peers fetch checkpoints every 2 steps by default	преди 3 години
callback.py	4918c58cb6 Polish stdout	преди 3 години
data.py	c61c61b20d Use t5-small tokenizer	преди 3 години
huggingface_auth.py	3e604bc1f5 fix auth	преди 3 години
manage_scaleset.py	c365b2ec9f Tweak settings for the upcoming demo (#2)	преди 3 години
requirements.txt	a2e1dfc253 Use user access token instead of username/password (#4)	преди 3 години
run_aux_peer.py	7ac2b44807 add `self.repo.git_pull` before `self.repo.push_to_hub` (#5)	преди 3 години
run_trainer.py	c365b2ec9f Tweak settings for the upcoming demo (#2)	преди 3 години
run_trainer_tpu.py	64dee420da Upgrade to using hivemind.optim.experimental	преди 3 години
task.py	2fed6ba68e Use HF username as wandb run name	преди 3 години
utils.py	f621362466 Make logging less verbose	преди 3 години

Training DALL-E with volunteers from all over the Internet

This repository is a part of the NeurIPS 2021 demonstration "Training Transformers Together".

In this demo, we train a model similar to OpenAI DALL-E — a Transformer "language model" that generates images from text descriptions. Training happens collaboratively — volunteers from all over the Internet contribute to the training using hardware available to them. We use LAION-400M, the world's largest openly available image-text-pair dataset with 400 million samples. Our model is based on the dalle‑pytorch implementation by Phil Wang with a few tweaks to make it communication-efficient.

See details about how to join and how it works on our website.

README.md

Training DALL-E with volunteers from all over the Internet