add tqdm

SMILELab-FL · Oct 11, 2021 · 659a270 · 659a270
1 parent 7743983
commit 659a270
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,7 @@
 numpy
 pandas
 pynvml
+tqdm
 
 torch>=1.7.1
 torchvision>=0.8.2
diff --git a/setup.py b/setup.py
@@ -26,7 +26,8 @@
                       'torchvision>=0.8.2',
                       'numpy',
                       'pandas',
-                      'pynvml'],
+                      'pynvml',
+                      'tqdm'],
     python_requires='>=3.6',
     classifiers=[
         'Topic :: Scientific/Engineering',