/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
python/cpp_bindings/adam_py.cpp
19 строк
571 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/adam.hpp> namespace py = pybind11; using namespace adept; void bind_adam(py::module& m) { py::class_<Adam>(m, "Adam") .def(py::init<std::vector<Variable>, float32_t, float32_t, float32_t, float32_t>(), py::arg("params"), py::arg("lr") = 0.001f, py::arg("beta1") = 0.9f, py::arg("beta2") = 0.999f, py::arg("eps") = 1e-8f) .def("step", &Adam::step) .def("zero_grad", &Adam::zero_grad) .def("set_lr", &Adam::set_lr) .def("lr", &Adam::lr); }