Struct AdamWParamState

• Defined in File adamw.h

Inheritance Relationships

Base Type

• public torch::optim::OptimizerCloneableParamState< AdamWParamState > (Template Class OptimizerCloneableParamState)

Struct Documentation

struct torch::optim::AdamWParamState : public torch::optim::OptimizerCloneableParamState<AdamWParamState>

Public Functions

auto step(const int64_t &new_step) -> decltype(*this)

auto step(int64_t &&new_step) -> decltype(*this)

const int64_t &step() const noexcept

int64_t &step() noexcept

auto exp_avg(const torch::Tensor &new_exp_avg) -> decltype(*this)

auto exp_avg(torch::Tensor &&new_exp_avg) -> decltype(*this)

const torch::Tensor &exp_avg() const noexcept

torch::Tensor &exp_avg() noexcept

auto exp_avg_sq(const torch::Tensor &new_exp_avg_sq) -> decltype(*this)

auto exp_avg_sq(torch::Tensor &&new_exp_avg_sq) -> decltype(*this)

const torch::Tensor &exp_avg_sq() const noexcept

torch::Tensor &exp_avg_sq() noexcept

auto max_exp_avg_sq(const torch::Tensor &new_max_exp_avg_sq) -> decltype(*this)

auto max_exp_avg_sq(torch::Tensor &&new_max_exp_avg_sq) -> decltype(*this)

const torch::Tensor &max_exp_avg_sq() const noexcept

torch::Tensor &max_exp_avg_sq() noexcept

void serialize(torch::serialize::InputArchive &archive) override

void serialize(torch::serialize::OutputArchive &archive) const override

~AdamWParamState() override = default

Friends

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