-
Notifications
You must be signed in to change notification settings - Fork 51
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ENH][REF] Torch Scoreboard #612
Conversation
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## torch #612 +/- ##
========================================
Coverage ? 92.01%
========================================
Files ? 128
Lines ? 11515
Branches ? 0
========================================
Hits ? 10596
Misses ? 919
Partials ? 0
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. |
tried to build an onnx model from the new scoreboard model using old encoder code in the notebook, first error was to put some variables on GPU, now the error is |
how are you defining the amps tensor? i've made some changes that may bring it back in line with the encoder from the colab notebook |
Description
Adds a PyTorch backend/engine for the cortex ScoreboardModel. This backend can be selected by passing the argument
engine='torch'
during model initialization.This also adds the
TorchBaseModel
abstract class which all torch models will subclass.Closes #611
Type of Change
Please delete options that are not relevant.
Checklist