diff --git a/requirements.txt b/requirements.txt index 4f46d259..ad5b5702 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ numpy opencv-python pyyaml scipy -tb-nightly +tensorboard torch>=1.7 torchvision tqdm