jactorch.optim.custom_optimizer_base#

Classes

CustomizedOptimizer

Base class for all customized optimizers.

Class CustomizedOptimizer

class CustomizedOptimizer[source]#

Bases: object

Base class for all customized optimizers.

__init__()#

__new__(**kwargs)#

load_state_dict(state_dict)[source]#: Load the state of the optimizer from a dictionary.

state_dict()[source]#: A dictionary that contains the state of the optimizer.

step(closure=None)[source]#: Performs a single optimization step.

zero_grad()[source]#: Clear the gradients of all optimized parameters.

property param_groups#: The parameter groups of the optimizer.

property state#: The state of the optimizer.