Update resnet_lightning.py

resnet_lightning.py

@@ -9,16 +9,13 @@ import torch.nn as nn
 import torch.nn.functional as F
 import torch.optim as optim
 from torchvision import datasets
-
-
 import albumentations as A
 from albumentations.pytorch import ToTensorV2
-
-
 from torch_lr_finder import LRFinder
 import lightning as L
 from pytorch_grad_cam import GradCAM
 from utils import *
+import torchmetrics
 
 class BasicBlock(nn.Module):
     expansion = 1