Class Adam
Defined in File adam.h
Page Contents
Inheritance Relationships
Base Type
public torch::optim::Optimizer
(Class Optimizer)
Class Documentation
-
class Adam : public torch::optim::Optimizer
Public Functions
-
inline explicit Adam(const std::vector<OptimizerParamGroup> ¶m_groups, AdamOptions defaults = {})
-
inline explicit Adam(std::vector<Tensor> params, AdamOptions defaults = {})
-
virtual torch::Tensor step(LossClosure closure = nullptr) override
A loss function closure, which is expected to return the loss value.
-
virtual void save(serialize::OutputArchive &archive) const override
Serializes the optimizer state into the given
archive
.
-
virtual void load(serialize::InputArchive &archive) override
Deserializes the optimizer state from the given
archive
.
-
inline explicit Adam(const std::vector<OptimizerParamGroup> ¶m_groups, AdamOptions defaults = {})