Class AdamW¶
Defined in File adamw.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::optim::Optimizer
(Class Optimizer)
Class Documentation¶
-
class
torch::optim
::
AdamW
: public torch::optim::Optimizer¶ Public Functions
-
AdamW
(std::vector<OptimizerParamGroup> param_groups, AdamWOptions defaults = {})¶
-
AdamW
(std::vector<Tensor> params, AdamWOptions defaults = {})¶
-
torch::Tensor
step
(LossClosure closure = nullptr) override¶ A loss function closure, which is expected to return the loss value.
-