Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
optimizer.SgdOptimizer Class Reference
Inheritance diagram for optimizer.SgdOptimizer:
optimizer.Optimizer

Public Member Functions

def __init__ (self, base_learning_rate=0.01, policy='fixed', momentum=0.0, kwargs)
 
def __call__ (self, net, param_init_net, param, grad)
 
- Public Member Functions inherited from optimizer.Optimizer
def __init__ (self)
 
def __call__ (self, net, param_init_net, param, grad)
 
def get_auxiliary_parameters (self)
 

Public Attributes

 base_learning_rate
 
 policy
 
 momentum
 
 init_kwargs
 

Additional Inherited Members

- Static Public Member Functions inherited from optimizer.Optimizer
def build_lr (net, param_init_net, base_learning_rate, learning_rate_blob="lr", policy="fixed", iter_val=0, kwargs)
 
def dedup (net, sparse_dedup_aggregator, grad)
 

Detailed Description

Definition at line 80 of file optimizer.py.


The documentation for this class was generated from the following file: