collection of experiments for DiT vs UNet pretraining of diffusion models

Name		Name	Last commit message	Last commit date
Latest commit History 26 Commits
utils		utils
.gitignore		.gitignore
ReadMe.md		ReadMe.md
common.py		common.py
dit.py		dit.py
get-pip.py		get-pip.py
imagenet.py		imagenet.py
imagenet256_fidstats_openai.npz		imagenet256_fidstats_openai.npz
imagenet_labels.txt		imagenet_labels.txt
quant_utils.py		quant_utils.py
requirements.txt		requirements.txt
setup.sh		setup.sh
shampoo.py		shampoo.py
sweep.sh		sweep.sh
test_tpu.py		test_tpu.py
train.py		train.py
unet.py		unet.py
unet_old.py		unet_old.py
vae-test.ipynb		vae-test.ipynb

Provide feedback