Skip to content

Conversation

@UrjaPawar
Copy link

If 'yss' is output of a pytorch model, then this check is necessary in order to run a pytorch model trained on tabular dataset: if not isinstance(yss, np.ndarray):
yss = yss.detach().numpy()

If 'yss' is output of a pytorch model, then this check is necessary in order to run a pytorch model trained on tabular dataset:
if not isinstance(yss, np.ndarray):
     yss = yss.detach().numpy()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant