Draft: Learning with backend cuda

Context

Support backend cuda for learning

Modified files

  • SGD.hpp, make SGD optimizer support backend_cuda;

TODO

  • Support backend_cuda for SGD
  • Support Accuracy computation on backend_cuda
Edited by Houssem ROUIS

Merge request reports

Loading