ONE - On-device Neural Engine
|
Public Member Functions | |
__init__ (self, learning_rate=0.001, nums_trainable_ops=trainable_ops.ALL) | |
Data Fields | |
learning_rate | |
nums_trainable_ops | |
Base class for optimizers.
Definition at line 4 of file optimizer.py.
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.
optimizer.optimizer.Optimizer.learning_rate |
Definition at line 14 of file optimizer.py.
optimizer.optimizer.Optimizer.nums_trainable_ops |
Definition at line 15 of file optimizer.py.