Update resnet.py

resnet.py

@@ -8,6 +8,23 @@ Reference:
 """
 import torch.nn as nn
 import torch.nn.functional as F
+# imports
+import os
+
+import torch
+from pytorch_lightning import LightningModule, Trainer
+from torch import nn
+from torch.nn import functional as F
+from torch.utils.data import DataLoader, random_split
+from torchmetrics import Accuracy
+from torchvision import transforms
+from torchvision.datasets import CIFAR10
+# from pytorch_lightning.callbacks import ModelSummary
+# from lightning.pytorch.callbacks import ModelCheckpoint
+from pytorch_lightning.callbacks import ModelCheckpoint, ModelSummary
+import albumentations as A
+from albumentations.pytorch import ToTensorV2
+import torchvision.transforms as transforms
 
 
 class BasicBlock(nn.Module):