diff --git a/requirements.txt b/requirements.txt index 0649f5bf..f7651bd1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,6 +5,7 @@ tqdm scikit-learn munch seaborn +matplotlib torch>=1.7.1 torchvision>=0.8.2 \ No newline at end of file