Utilities for Neural Network training
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
valbub opened this issue 5 years ago · comments
When validating with metric that returns a GPU tensor, you get an error. Scorch tries to call metric(...).numpy(), but has to call metric(...).cpu().numpy().