Skip to content

Draft: Learning with backend cuda

Houssem ROUIS requested to merge hrouis/aidge_learning:backend_cuda into dev

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