requirements.txt 68 Bytes
Newer Older
zbian's avatar
zbian committed
1
2
3
4
5
6
7
torch>=1.8
torchvision>=0.9
numpy
tqdm
psutil
tensorboardX
packaging