Update resnet.py

resnet.py

@@ -1,11 +1,11 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-#from torchsummary import summary
+from torchsummary import summary
 from io import BytesIO
 import numpy as np
 import os
-#from pytorch_lightning import LightningModule, Trainer
+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