update resnet

resnet.py

@@ -46,7 +46,6 @@ class custom_ResNet(pl.LightningModule):
       # Set our init args as class attributes
       # Hardcode some dataset specific attributes
         self.data_dir = data_dir
-        self.learning_rate = learning_rate
         self.classes = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
         self.num_classes = 10
         self.train_transform = transforms.Compose([