Skip to content

Commit 99a4936

Browse files
authored
Merge pull request #29 from mskcc/remove_torchvision
remove torchvision requirement
2 parents 8d1a20e + cc3c682 commit 99a4936

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

data/data_loader.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
from torch.utils.data import Dataset, DataLoader
2121
import os
2222
from sklearn.preprocessing import normalize
23-
from torchvision import transforms
2423

2524

2625
class MSIBags(Dataset):

requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,5 @@ scikit-learn
1010
scipy>=1.1.0; python_version <= '2.7'
1111
scipy
1212
torch
13-
torchvision
1413
matplotlib
1514

0 commit comments

Comments
 (0)