Merge pull request #29 from mskcc/remove_torchvision

remove torchvision requirement
mskcc · Aug 27, 2021 · 99a4936 · 99a4936
2 parents 8d1a20e + cc3c682
commit 99a4936
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/data/data_loader.py b/data/data_loader.py
@@ -20,7 +20,6 @@
 from torch.utils.data import Dataset, DataLoader
 import os
 from sklearn.preprocessing import normalize
-from torchvision import transforms
 
 
 class MSIBags(Dataset):

diff --git a/requirements.txt b/requirements.txt
@@ -10,6 +10,5 @@ scikit-learn
 scipy>=1.1.0; python_version <= '2.7'
 scipy
 torch
-torchvision
 matplotlib