75ae7742ab
For some strategies we don't do all reduce, so all_reduce_sum_gradients can be misleading. The parameter is also changed to experimental because of issues with CentralStorageStrategy. PiperOrigin-RevId: 302734837 Change-Id: Ic30e2f81ab61eef568ee68e5752015f950117d47
72 lines
2.8 KiB
Plaintext
72 lines
2.8 KiB
Plaintext
path: "tensorflow.optimizers.Adam"
|
|
tf_class {
|
|
is_instance: "<class \'tensorflow.python.keras.optimizer_v2.adam.Adam\'>"
|
|
is_instance: "<class \'tensorflow.python.keras.optimizer_v2.optimizer_v2.OptimizerV2\'>"
|
|
is_instance: "<class \'tensorflow.python.training.tracking.base.Trackable\'>"
|
|
is_instance: "<type \'object\'>"
|
|
member {
|
|
name: "iterations"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "weights"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member_method {
|
|
name: "__init__"
|
|
argspec: "args=[\'self\', \'learning_rate\', \'beta_1\', \'beta_2\', \'epsilon\', \'amsgrad\', \'name\'], varargs=None, keywords=kwargs, defaults=[\'0.001\', \'0.9\', \'0.999\', \'1e-07\', \'False\', \'Adam\'], "
|
|
}
|
|
member_method {
|
|
name: "add_slot"
|
|
argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\'], varargs=None, keywords=None, defaults=[\'zeros\'], "
|
|
}
|
|
member_method {
|
|
name: "add_weight"
|
|
argspec: "args=[\'self\', \'name\', \'shape\', \'dtype\', \'initializer\', \'trainable\', \'synchronization\', \'aggregation\'], varargs=None, keywords=None, defaults=[\'None\', \'zeros\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
|
|
}
|
|
member_method {
|
|
name: "apply_gradients"
|
|
argspec: "args=[\'self\', \'grads_and_vars\', \'name\', \'experimental_aggregate_gradients\'], varargs=None, keywords=None, defaults=[\'None\', \'True\'], "
|
|
}
|
|
member_method {
|
|
name: "from_config"
|
|
argspec: "args=[\'cls\', \'config\', \'custom_objects\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "get_config"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_gradients"
|
|
argspec: "args=[\'self\', \'loss\', \'params\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_slot"
|
|
argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_slot_names"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_updates"
|
|
argspec: "args=[\'self\', \'loss\', \'params\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_weights"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "minimize"
|
|
argspec: "args=[\'self\', \'loss\', \'var_list\', \'grad_loss\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "set_weights"
|
|
argspec: "args=[\'self\', \'weights\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "variables"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
}
|