BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
|
#include <stochastic_gradient_descent.h>
Public Types | |
using | value_type = bc::traits::conditional_detected_t< bc::traits::query_value_type, ValueType, ValueType > |
Public Member Functions | |
template<class... Args> | |
Optimizer (Args &&...) | |
template<class TensorX , class Gradients > | |
void | update (TensorX &tensor, Gradients &&delta) |
void | set_learning_rate (value_type lr) |
![]() | |
void | save (Layer_Loader &loader, std::string name) const |
void | load (Layer_Loader &loader, std::string name) |
Public Attributes | |
value_type | learning_rate = 0.003 |
using bc::nn::Stochastic_Gradient_Descent::Optimizer< ValueType >::value_type = bc::traits::conditional_detected_t< bc::traits::query_value_type, ValueType, ValueType> |
|
inline |
|
inline |
|
inline |
value_type bc::nn::Stochastic_Gradient_Descent::Optimizer< ValueType >::learning_rate = 0.003 |