padmanabhbosamia
commited on
Commit
·
eae1260
1
Parent(s):
d94c842
Update resnet.py
Browse files
resnet.py
CHANGED
@@ -9,7 +9,7 @@ from pytorch_lightning import LightningModule, Trainer
|
|
9 |
from torch import nn
|
10 |
from torch.nn import functional as F
|
11 |
from torch.utils.data import DataLoader, random_split
|
12 |
-
from torchmetrics import Accuracy
|
13 |
from torchvision import transforms
|
14 |
from torchvision.datasets import CIFAR10
|
15 |
from torch_lr_finder import LRFinder
|
|
|
9 |
from torch import nn
|
10 |
from torch.nn import functional as F
|
11 |
from torch.utils.data import DataLoader, random_split
|
12 |
+
#from torchmetrics import Accuracy
|
13 |
from torchvision import transforms
|
14 |
from torchvision.datasets import CIFAR10
|
15 |
from torch_lr_finder import LRFinder
|