Shortcuts

Struct AdamWOptions

Inheritance Relationships

Base Type

Struct Documentation

struct AdamWOptions : public torch::optim::OptimizerCloneableOptions<AdamWOptions>

Public Functions

AdamWOptions(double lr = 1e-3)
inline auto lr(const double &new_lr) -> decltype(*this)
inline auto lr(double &&new_lr) -> decltype(*this)
inline const double &lr() const noexcept
inline double &lr() noexcept
inline auto betas(const betas_t &new_betas) -> decltype(*this)
inline auto betas(betas_t &&new_betas) -> decltype(*this)
inline const betas_t &betas() const noexcept
inline betas_t &betas() noexcept
inline auto eps(const double &new_eps) -> decltype(*this)
inline auto eps(double &&new_eps) -> decltype(*this)
inline const double &eps() const noexcept
inline double &eps() noexcept
inline auto weight_decay(const double &new_weight_decay) -> decltype(*this)
inline auto weight_decay(double &&new_weight_decay) -> decltype(*this)
inline const double &weight_decay() const noexcept
inline double &weight_decay() noexcept
inline auto amsgrad(const bool &new_amsgrad) -> decltype(*this)
inline auto amsgrad(bool &&new_amsgrad) -> decltype(*this)
inline const bool &amsgrad() const noexcept
inline bool &amsgrad() noexcept
virtual void serialize(torch::serialize::InputArchive &archive) override
virtual void serialize(torch::serialize::OutputArchive &archive) const override
virtual double get_lr() const override
virtual void set_lr(const double lr) override

Friends

friend bool operator==(const AdamWOptions &lhs, const AdamWOptions &rhs)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources