Skip to content

Learning backend cuda

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

Context

Support backend cuda for learning

Modified files

  • SGD.hpp, make SGD optimizer support backend_cuda;
  • Accuracy.hpp, Accuracy.cpp and pybind_Accuracy.hpp add Accuracy metric graph;

TODO

  • Support backend_cuda for SGD
  • Support Accuracy computation on backend_cuda

Merge request reports

Loading