'Serializes and saves model parameters to the path specified.
Arguments:
param_path (str): File to write serialized parameter dict to.
keep_states (bool): Whether to save optimizer states too.
Defaults to True.'

def save_params(self, param_path, keep_states=True):

    self.serialize(keep_states=keep_states, fn=param_path)
