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.
2 parents f386ca0 + 1319ea3 commit 57cb73bCopy full SHA for 57cb73b
art/estimators/regression/pytorch.py
@@ -269,7 +269,7 @@ def predict( # pylint: disable=W0221
269
output = model_outputs[-1]
270
output = output.detach().cpu().numpy().astype(np.float32)
271
if len(output.shape) == 1:
272
- output = np.expand_dims(output.detach().cpu().numpy(), axis=1).astype(np.float32)
+ output = np.expand_dims(output, axis=1).astype(np.float32)
273
274
results_list.append(output)
275
0 commit comments