MCPcopy Create free account
hub / github.com/DeepGraphLearning/graphvite / RMSprop

Method RMSprop

include/core/optimizer.h:301–306  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

299class RMSprop : public Optimizer {
300public:
301 RMSprop(float _lr = 1e-4, float _weight_decay = 0, float _alpha = 0.999, float _epsilon = 1e-8,
302 const LRSchedule &_schedule = "linear") :
303 Optimizer("RMSprop", 1, _lr, _weight_decay, _schedule) {
304 alpha = _alpha;
305 epsilon = _epsilon;
306 }
307};
308
309/** Helper class for Adam */

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected