#include <optimizer.hpp>
|
| sgd (Loss &loss, std::size_t batch_size, T learning_rate=1.0e-1, T momentum=0.0, T decay=0.0, bool nesterov=false) noexcept |
|
void | forward () |
|
◆ tensor_type
template<typename Loss , typename T >
◆ sgd()
template<typename Loss , typename T >
ceras::sgd< Loss, T >::sgd |
( |
Loss & |
loss, |
|
|
std::size_t |
batch_size, |
|
|
T |
learning_rate = 1.0e-1 , |
|
|
T |
momentum = 0.0 , |
|
|
T |
decay = 0.0 , |
|
|
bool |
nesterov = false |
|
) |
| |
|
inlinenoexcept |
◆ forward()
template<typename Loss , typename T >
◆ decay_
template<typename Loss , typename T >
◆ iterations_
template<typename Loss , typename T >
◆ learning_rate_
template<typename Loss , typename T >
◆ loss_
template<typename Loss , typename T >
◆ momentum_
template<typename Loss , typename T >
◆ nesterov_
template<typename Loss , typename T >
The documentation for this struct was generated from the following file: