| --extra-index-url https://download.pytorch.org/whl/cu113 | |
| torch | |
| torchvision | |
| einops | |
| ema-pytorch | |
| fsspec | |
| fvcore | |
| huggingface-hub | |
| matplotlib | |
| numpy==1.26.4 | |
| opencv-python | |
| omegaconf | |
| pytorch-msssim | |
| scikit-image | |
| scikit-learn | |
| scipy | |
| tensorboard | |
| tensorboardx | |
| wandb | |
| timm | |