Struct AdamWParamState¶
Defined in File adamw.h
Page Contents
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)¶
-
auto