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

Public Member Functions

def __init__ (self, base_learning_rate=0.1, momentum=0.0, policy="fixed", nesterov=1, weight_decay=0.0001, sparse_dedup_aggregator=None, kwargs)
- Public Member Functions inherited from caffe2.python.optimizer.SgdOptimizer
def __init__ (self, base_learning_rate=0.01, policy='fixed', momentum=0.0, nesterov=1, sparse_dedup_aggregator=None, lars=None, kwargs)
def scale_learning_rate (self, scale)
- Public Member Functions inherited from caffe2.python.optimizer.Optimizer
def __init__ (self)
def __call__ (self, net, param_init_net, param, grad=None)
def get_cpu_blob_name (self, base_str, node_name='')
def get_gpu_blob_name (self, base_str, gpu_id, node_name)
def make_unique_blob_name (self, base_str)
def build_lr (self, net, param_init_net, base_learning_rate, learning_rate_blob=None, policy="fixed", iter_val=0, kwargs)
def add_lr_multiplier (self, lr_multiplier)
def get_auxiliary_parameters (self)
def scale_learning_rate (self, args, kwargs)
def create_lars_inputs (self, param_init_net, weight_decay, trust, lr_max)

Public Attributes

- Public Attributes inherited from caffe2.python.optimizer.SgdOptimizer

Additional Inherited Members

- Static Public Member Functions inherited from caffe2.python.optimizer.Optimizer
def dedup (net, sparse_dedup_aggregator, grad)

Detailed Description

Definition at line 384 of file

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