(self, optimization_parameters)
| 1241 | """Handles Adam specific logic.""" |
| 1242 | |
| 1243 | def __init__(self, optimization_parameters): |
| 1244 | super(_AdamHandler, self).__init__(optimization_parameters) |
| 1245 | self._table_to_m_variables_dict = {} |
| 1246 | self._table_to_v_variables_dict = {} |
| 1247 | |
| 1248 | def set_optimization_parameters(self, table_descriptor): |
| 1249 | table_descriptor.optimization_parameters.adam.beta1 = ( |