update requirements.txt

ant-research · Sep 11, 2020 · 9756789 · 9756789
1 parent d6cee6e
commit 9756789
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -8,3 +8,5 @@ scikit-learn==0.22.1
 torch==1.4.0
 torchvision==0.5.0
 tqdm==4.42.1
+albumentations==0.4.5
+pretrainedmodels==0.7.4