Defined in File adamw.h
AdamW: public torch::optim::Optimizer¶
step(LossClosure closure = nullptr) override¶
A loss function closure, which is expected to return the loss value.
save(serialize::OutputArchive &archive) const override¶
Serializes the optimizer state into the given