|
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) |
Public Member Functions inherited from bc::nn::Optimizer_Base | |
| 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 |
1.8.13