We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 036cc9f commit a480e17Copy full SHA for a480e17
art/estimators/classification/pytorch.py
@@ -951,7 +951,6 @@ def __repr__(self):
951
952
def _make_model_wrapper(self, model: "torch.nn.Module") -> "torch.nn.Module":
953
# 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
955
try:
956
import torch.nn as nn
957
0 commit comments