padmanabhbosamia commited on
Commit
eae1260
·
1 Parent(s): d94c842

Update resnet.py

Browse files
Files changed (1) hide show
  1. resnet.py +1 -1
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