requirements.txt - diffusion-mnist - Diffusion-based generative experiments for MNIST digits.

about summary refs log tree commit diff

path: root/requirements.txt

blob: d590e96b97e12f57da61b988a9ed0b2aad25ba2d (plain) (blame)

torchvision==0.15.2
matplotlib==3.8.0
rich==13.3.5
PyQt5==5.15.10
torch-trainer @ git+https://github.com/gdamms/torch-trainer.git@da70ca78adf6195f2e3add92938b9c282d5c94cb