def set_shared_states(a, b):
assert isinstance(a, chainer.Optimizer)
assert hasattr(a, 'target'), 'Optimizer.setup must be called first'
for param_name, param in a.target.namedparams():
ensure_initialized_update_rule(param)
state = param.update_rule.state
for state_name, state_val in b[param_name].items():
s = state[state_name]
state[state_name] = np.frombuffer(
state_val,
dtype=s.dtype).reshape(s.shape)
评论列表
文章目录