automl / automl/TransformersCanDoBayesianInference

Error found in the batch_pred function

Open
#3 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
272
Forks
41
PR merge metrics
No merged PRs in 30d

Description

Error message:
[Cannot convert list to array: ValueError: only one element tensors can be converted to Python scalars]

In tabular.py,
output = output.detach().cpu().numpy() in the batch_pred function.
Otherwise, it produces an error when running BayesianNN model.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.