/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
python/cpp_bindings/sgd_py.cpp
17 строк
389 B
kolkir
Refactor python folder structure
01 мар 2025, 22:16
01 мар 2025, 22:16
ef25f05
Код
Авторство
О чём код?
#include <pybind11/pybind11.h> #include <pybind11/stl.h> #include <adept/nn/sgd.hpp> namespace py = pybind11; using namespace adept; void bind_sgd(py::module& m) { py::class_<SGD>(m, "SGD") .def(py::init<std::vector<Variable>, float32_t>()) .def("step", &SGD::step) .def("zero_grad", &SGD::zero_grad) .def("set_lr", &SGD::set_lr) .def("lr", &SGD::lr); }