 |
Caffe2 - Python API
A deep learning, cross platform ML framework
|
|
Caffe2 - Python API
A deep learning, cross platform ML framework
|
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 | |
| weight_decay | |
| base_learning_rate | |
| Public Attributes inherited from caffe2.python.optimizer.SgdOptimizer | |
| base_learning_rate | |
| policy | |
| momentum | |
| nesterov | |
| sparse_dedup_aggregator | |
| lars | |
| init_kwargs | |
Additional Inherited Members | |
| Static Public Member Functions inherited from caffe2.python.optimizer.Optimizer | |
| def | dedup (net, sparse_dedup_aggregator, grad) |
Definition at line 384 of file optimizer.py.
1.8.11
