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 optinal torch-based solver #73

Open
tomMoral opened this issue Mar 10, 2022 · 1 comment
Open

ENH add optinal torch-based solver #73

tomMoral opened this issue Mar 10, 2022 · 1 comment

Comments

@tomMoral
Copy link
Contributor

In our recent paper with @Malezieux, we achieved faster performances with almost no performance drop with a torch-based algorithm.
It would be nice to implement such algorithm in alphcsc, with an optional dependency on pytorch.

Where to add this algorithm is not totally straightforward. I think the easier would probably be to implement a separate algorithm similar to _batch_learn and _online_learn. But another possibility would be to refactor the code to better take into account the bi-level optimization framework (less alternate minimization), which would ease this integration (and follow up that will unfold from our work on the field).

@agramfort
Copy link
Collaborator

agramfort commented Mar 10, 2022 via email

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

2 participants