Update custom_resnet.py

custom_resnet.py

@@ -1,3 +1,13 @@
+from pytorch_lightning import LightningModule, Trainer, seed_everything
+import torch
+import pandas as pd
+import numpy as np
+import torch.nn as nn
+from torch.nn import functional as F
+from torchvision import transforms
+from pytorch_lightning import LightningModule, Trainer, seed_everything
+
+
 class LitResnet(LightningModule):
     def __init__(self, lr=0.01, drop= 0.2, norm='BN',groupsize=1):
         super().__init__()