helper.py 文件源码

def save(self, file_path, **kwargs):
    """
    save variables to file (using pickle)
    """
    #check if any variable is a dict
    var_count = 0
    for key in kwargs:
      var_count += 1
      if isinstance(kwargs[key],dict):
        sys.stderr.write('Opps! Cannot write a dictionary into pickle')
        sys.exit(1)
    with open(file_path,'wb') as f:
      pickler = pickle.Pickler(f,-1)
      pickler.dump(var_count)
      for key in kwargs:
        pickler.dump(key)
        pickler.dump(kwargs[key])