Skip to content

Commit 972a743

Browse files
committed
mxnet fix bug in predict
1 parent 2dd43d7 commit 972a743

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

art/classifiers/mxnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def predict(self, x, logits=False, batch_size=128):
131131
with autograd.record(train_mode=False):
132132
preds = self._model(x_batch)
133133

134-
if logits is True:
134+
if logits is False:
135135
preds = preds.softmax()
136136

137137
# preds = np.empty((x.shape[0], self.nb_classes), dtype=float)

0 commit comments

Comments
 (0)