Skip to content

Commit a480e17

Browse files
author
TrojAISec
committed
remove unneccessary nb_classes var
1 parent 036cc9f commit a480e17

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

art/estimators/classification/pytorch.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -951,7 +951,6 @@ def __repr__(self):
951951

952952
def _make_model_wrapper(self, model: "torch.nn.Module") -> "torch.nn.Module":
953953
# Try to import PyTorch and create an internal class that acts like a model wrapper extending torch.nn.Module
954-
nb_classes = self.nb_classes
955954
try:
956955
import torch.nn as nn
957956

0 commit comments

Comments
 (0)