code: def _set_state_impl(self, key, param):
      self.m[key] = np.zeros_like(param)
      self.s[key] = np.zeros_like(param)
      self.t[key] = 0
      # s_max is not always necessary but is set for keeping the code simple
      self.s_max[key] = np.zeros_like(param)
  
file_path: nnabla/python/test/solver/test_adabelief.py
imports: import pytest
  import nnabla.solvers
  import numpy
  from solver_test_utils import solver_tester, RefSolver
  from nbla_test_utils import list_context
label_desc: Use image node_id 3 for calling the RefAdaBelief obj's underlying member method code with example usage: obj._set_state_impl(key, param) without return types