ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
optimizer.optimizer.Optimizer Class Reference

Public Member Functions

 __init__ (self, learning_rate=0.001, nums_trainable_ops=trainable_ops.ALL)
 

Data Fields

 learning_rate
 
 nums_trainable_ops
 

Detailed Description

Base class for optimizers.

Definition at line 4 of file optimizer.py.

Constructor & Destructor Documentation

◆ __init__()

optimizer.optimizer.Optimizer.__init__ (   self,
  learning_rate = 0.001,
  nums_trainable_ops = trainable_ops.ALL 
)
Initialize the optimizer.
Args:
    learning_rate (float): The learning rate for optimization.

Reimplemented in optimizer.adam.Adam, and optimizer.sgd.SGD.

Definition at line 8 of file optimizer.py.

8 def __init__(self, learning_rate=0.001, nums_trainable_ops=trainable_ops.ALL):
9 """
10 Initialize the optimizer.
11 Args:
12 learning_rate (float): The learning rate for optimization.
13 """
14 self.learning_rate = learning_rate
15 self.nums_trainable_ops = nums_trainable_ops

Field Documentation

◆ learning_rate

optimizer.optimizer.Optimizer.learning_rate

Definition at line 14 of file optimizer.py.

◆ nums_trainable_ops

optimizer.optimizer.Optimizer.nums_trainable_ops

Definition at line 15 of file optimizer.py.


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