Make deep methods work on GPU

scikit-adaptation / skada

Domain adaptation toolbox compatible with scikit-learn and pytorch

https://scikit-adaptation.github.io/

BSD 3-Clause "New" or "Revised" License

54 stars 16 forks source link

Make deep methods work on GPU #120

Open tgnassou opened 4 months ago

tgnassou commented 4 months ago

Deep methods do not currently support device='gpu'. It should break at several locations when we define a new tensor or a domain_classifier for the adversarial methods. This issue should not be too hard to fix.