Update resnet_lightning.py

resnet_lightning.py

@@ -77,8 +77,7 @@ class ResNet18Model(L.LightningModule):
         self.layer3 = self._make_layer(block, 256, num_blocks[2], stride=2)
         self.layer4 = self._make_layer(block, 512, num_blocks[3], stride=2)
         self.linear = nn.Linear(512*block.expansion, num_classes)
-
-        self.accuracy = Accuracy(task="multiclass", num_classes=10)
+        self.accuracy = torchmetrics.classification.Accuracy(task="multiclass", num_classes=10)
 
     def _make_layer(self, block, planes, num_blocks, stride):
         strides = [stride] + [1]*(num_blocks-1)