Public Member Functions | |
template<typename ParameterContainer > | |
Adagrad (ParameterContainer &¶meters, const AdagradOptions &options) | |
void | step () override |
Adapted from https://github.com/pytorch/pytorch/blob/master/torch/optim/adagrad.py. | |
void | save (serialize::OutputArchive &archive) const override |
Serializes the optimizer state into the given archive . | |
void | load (serialize::InputArchive &archive) override |
Deserializes the optimizer state from the given archive . | |
Public Member Functions inherited from torch::optim::detail::OptimizerBase | |
OptimizerBase (std::vector< Tensor > parameters) | |
Constructs the Optimizer from a vector of parameters. | |
void | add_parameters (const std::vector< Tensor > ¶meters) |
Adds the given vector of parameters to the optimizer's parameter list. | |
virtual void | zero_grad () |
Zeros out the gradients of all parameters. | |
const std::vector< Tensor > & | parameters () const noexcept |
Provides a const reference to the parameters this optimizer holds. | |
std::vector< Tensor > & | parameters () noexcept |
Provides a reference to the parameters this optimizer holds. | |
size_t | size () const noexcept |
Returns the number of parameters referenced by the optimizer. | |
Data Fields | |
AdagradOptions | options |
std::vector< Tensor > | sum_buffers |
std::vector< int64_t > | step_buffers |
Additional Inherited Members | |
Protected Member Functions inherited from torch::optim::detail::OptimizerBase | |
template<typename T > | |
T & | buffer_at (std::vector< T > &buffers, size_t index) |
Accesses a buffer at the given index. More... | |
Tensor & | buffer_at (std::vector< Tensor > &buffers, size_t index) |
Accesses a buffer at the given index, converts it to the type of the parameter at the corresponding index (a no-op if they match). More... | |
Protected Attributes inherited from torch::optim::detail::OptimizerBase | |
std::vector< Tensor > | parameters_ |
The parameters this optimizer optimizes. | |