Rename train.py to train_proj.py
Browse files- train.py → train_proj.py +1 -1
train.py → train_proj.py
RENAMED
@@ -7,7 +7,7 @@ from torch import nn
|
|
7 |
from torch.utils.data import DataLoader
|
8 |
from torchvision import datasets
|
9 |
from torchvision.transforms import ToTensor, Normalize, RandomCrop, RandomHorizontalFlip, Compose
|
10 |
-
from
|
11 |
|
12 |
# data transforms
|
13 |
|
|
|
7 |
from torch.utils.data import DataLoader
|
8 |
from torchvision import datasets
|
9 |
from torchvision.transforms import ToTensor, Normalize, RandomCrop, RandomHorizontalFlip, Compose
|
10 |
+
from contextualizer_proj import Contextualizer
|
11 |
|
12 |
# data transforms
|
13 |
|