Skip to content

Commit 7eb6a8e

Browse files
Irina NicolaeIrina Nicolae
authored andcommitted
Remove commented batching code in MXNet
1 parent 972a743 commit 7eb6a8e

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

art/classifiers/mxnet.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -134,14 +134,6 @@ def predict(self, x, logits=False, batch_size=128):
134134
if logits is False:
135135
preds = preds.softmax()
136136

137-
# preds = np.empty((x.shape[0], self.nb_classes), dtype=float)
138-
# pred_iter = mx.io.NDArrayIter(data=x_, batch_size=128)
139-
# if logits is True:
140-
# for preds_i, i, batch in mod.iter_predict(pred_iter):
141-
# pred_label = preds_i[0].asnumpy()
142-
# else:
143-
# for preds_i, i, batch in mod.iter_predict(pred_iter):
144-
# pred_label = preds_i[0].softmax().asnumpy()
145137
results[begin:end] = preds.asnumpy()
146138

147139
return results

0 commit comments

Comments
 (0)