diff --git a/requirements.txt b/requirements.txt index e726dc0..2c7b2f6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,10 +1,10 @@ -glob imageio numpy -python-opencv-headless +opencv-python-headless pyyaml tensorboard -torch>1.3.0 -torchvision +torch>=1.3.0 +torchvision>=0.5 tqdm --e git://github.com/aliutkus/torchsearchsorted#egg=torchsearchsorted +git+https://github.com/aliutkus/torchsearchsorted +