# python=3.8
torch
torchvision
tensorboard
scikit-learn
matplotlib
numpy
einops
timm==0.4.12
tqdm
omega
pyyaml
opencv-python
wandb
icecream
torchinfo