(self, states)
| 327 | return states |
| 328 | |
| 329 | def set_states(self, states): |
| 330 | super().set_states(states) |
| 331 | if 'moments' in states: |
| 332 | self.moments = states['moments'] |
| 333 | self.mom_value = self.momentum(self.step_counter) |
| 334 | |
| 335 | |
| 336 | class RMSProp(Optimizer): |
no outgoing calls