#include <optimizer.hpp>
|
| adam (Loss &loss, std::size_t batch_size, T learning_rate=1.0e-1, T beta_1=0.9, T beta_2=0.999, bool amsgrad=false) noexcept |
|
void | forward () |
|
◆ tensor_type
template<typename Loss , typename T >
◆ adam()
template<typename Loss , typename T >
ceras::adam< Loss, T >::adam |
( |
Loss & |
loss, |
|
|
std::size_t |
batch_size, |
|
|
T |
learning_rate = 1.0e-1 , |
|
|
T |
beta_1 = 0.9 , |
|
|
T |
beta_2 = 0.999 , |
|
|
bool |
amsgrad = false |
|
) |
| |
|
inlinenoexcept |
◆ forward()
template<typename Loss , typename T >
◆ amsgrad_
template<typename Loss , typename T >
◆ beta_1_
template<typename Loss , typename T >
◆ beta_2_
template<typename Loss , typename T >
◆ iterations_
template<typename Loss , typename T >
◆ learning_rate_
template<typename Loss , typename T >
◆ loss_
template<typename Loss , typename T >
The documentation for this struct was generated from the following file: