Shortcuts

Class RMSprop

Inheritance Relationships

Base Type

Class Documentation

class RMSprop : public torch::optim::Optimizer

Public Functions

inline explicit RMSprop(std::vector<OptimizerParamGroup> param_groups, RMSpropOptions defaults = {})
inline explicit RMSprop(std::vector<Tensor> params, RMSpropOptions defaults = {})
virtual torch::Tensor step(LossClosure closure = nullptr) override

A loss function closure, which is expected to return the loss value.

virtual void save(serialize::OutputArchive &archive) const override

Serializes the optimizer state into the given archive.

virtual void load(serialize::InputArchive &archive) override

Deserializes the optimizer state from the given archive.

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