| torch>=2.3.0 | |
| torchvision | |
| torchaudio | |
| numpy | |
| matplotlib | |
| scikit-learn | |
| opencv-python | |
| einops>=0.7.0 | |
| omegaconf>=2.3.0 | |
| lightning>=2.1.3 | |
| transformers>=4.36.2 | |
| pudb | |
| wandb | |
| tensorboard | |
| termcolor | |
| lpips==0.1.4 | |
| ftfy | |
| imageio | |
| imageio-ffmpeg | |
| pyiqa | |
| clean-fid | |
| albumentations | |
| pytorch-fid | |
| torchinfo | |
| gradio |