Skip to content
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] Add torch engine for BiphasicAxonMapModel #615

Open
jgranley opened this issue Apr 15, 2024 · 0 comments
Open

[ENH] Add torch engine for BiphasicAxonMapModel #615

jgranley opened this issue Apr 15, 2024 · 0 comments
Assignees

Comments

@jgranley
Copy link
Member

jgranley commented Apr 15, 2024

Support engine='torch' for BiphasicAxonMapModel. The design should match as closely as possible the code for torch engine for #611 (in progress), meaning it should have a separate class that inherits from torches nn.Module, and predict_spatial is refactored to direct calls to the torch module if engine is torch. The torch module should have a function that takes in the stimulus, model parameters, and electrode locations (as torch tensors) and should return the percept as a torch tensor. All existing unit tests should pass.

If it is convenient then this could use the same torch module as AxonMapModel.

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

No branches or pull requests

3 participants