model.py 文件源码

def save_params(self, f_, format_=Default, filter_=None):
        '''
        This saves all parameters in current group, using pickle protocol

        An dict containing string->shared_variables will be dumped to f_.

        Args:
            f_: writable file or filename string
            format_: string, file format.
                Default is to interpret from file name
                supported format: "pkl"
            filter_: string or None, regex pattern to filter

        '''
        if isinstance(f_, str):
            f_ = open(f_, 'wb')
        if filter_ is None:
            pickle.dump(self._current_group_di, f_)
        else:
            pat = re.compile(filter_)
            pickle.dump({k:v for k,v in self._current_group_di.items() if pat.fullmatch(k)}, f_)