saver.py 文件源码

def __init__(self, folder, model_opt=None, data_opt=None):
    if not os.path.exists(folder):
      os.makedirs(folder)
    self.folder = folder
    self.log = logger.get()
    self.tf_saver = None
    if model_opt is not None:
      self.save_opt(os.path.join(folder, kModelOptFilename), model_opt)
    if data_opt is not None:
      self.save_opt(os.path.join(folder, kDatasetOptFilename), data_opt)