当前位置：网站首页>tf.train.MonitoredTrainingSession 控制 checkpoint 保存数量

tf.train.MonitoredTrainingSession 控制 checkpoint 保存数量

2022-08-08 06:20:00 【亓逸】

`tf.train.MonitoredTrainingSession` 控制 checkpoint 保存数量

分布式环境下，为了剩下存储ckpt保存数量磁盘空间

tf.train.MonitoredTrainingSession

控制checkpoint 数量

scaffold = tf.train.Scaffold(saver=tf.train.Saver(max_to_keep=1))

scaffold = tf.train.Scaffold(saver=tf.train.Saver(max_to_keep=1))
       self.sess = tf.train.MonitoredTrainingSession(
           master=self.server.target,
           checkpoint_dir=self.ckpt_dir,
           save_checkpoint_secs=30,
           is_chief=(self.task_index == 0),
           scaffold=scaffold,
           hooks=hooks,
           config=self.conf)