|
template<typename ParameterContainer > |
| Adam (ParameterContainer &¶meters, const AdamOptions &options) |
|
void | step () override |
|
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 .
|
|
| 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.
|
|
|
AdamOptions | options |
|
std::vector< int64_t > | step_buffers |
|
std::vector< Tensor > | exp_average_buffers |
|
std::vector< Tensor > | exp_average_sq_buffers |
|
std::vector< Tensor > | max_exp_average_sq_buffers |
|
|
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...
|
|
std::vector< Tensor > | parameters_ |
| The parameters this optimizer optimizes.
|
|
Definition at line 31 of file adam.h.
The documentation for this class was generated from the following files:
- torch/csrc/api/include/torch/optim/adam.h
- torch/csrc/api/src/optim/adam.cpp