File size: 152 Bytes
c608946
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
torch
einops
torchvision
opencv-python
kornia
albumentations
loguru
tqdm
matplotlib
h5py
wandb
timm
xformers # Optional, used for memefficient attention