Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
layers.optimizers.Optimizers Class Reference
Inheritance diagram for layers.optimizers.Optimizers:

Public Member Functions

def __init__ (self, optimizers)
 
def __enter__ (self)
 
def __exit__ (self, type, value, traceback)
 

Detailed Description

Optimizers context to provide optimizers to layers
within the context.

Example usage:
    optimizers = {'optim1': optim1, 'optim2': optim2}
    with Optimizers(optimizers):
        optim = OptimizerContext.current().get_optimizer('optim1')
        layer(optim=optim)

Definition at line 43 of file optimizers.py.


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