(self)
| 48 | self.checkpoint_dir = fs.normpath(checkpoint_dir) if checkpoint_dir is not None else checkpoint_dir |
| 49 | |
| 50 | def _setup_graph(self): |
| 51 | assert self.checkpoint_dir is not None, \ |
| 52 | "Please provide 'checkpoint_dir' for ModelSaver, or use logger.set_logger_dir()" |
| 53 | vars = [] |
| 54 | for key in self.var_collections: |
| 55 | vars.extend(tf.get_collection(key)) |
| 56 | vars = list(set(vars)) |
| 57 | self.path = os.path.join(self.checkpoint_dir, 'model') |
| 58 | self.saver = tf.train.Saver( |
| 59 | var_list=vars, |
| 60 | max_to_keep=self._max_to_keep, |
| 61 | keep_checkpoint_every_n_hours=self._keep_every_n_hours, |
| 62 | write_version=tf.train.SaverDef.V2, |
| 63 | save_relative_paths=True) |
| 64 | # Scaffold will call saver.build from this collection |
| 65 | tf.add_to_collection(tf.GraphKeys.SAVERS, self.saver) |
| 66 | |
| 67 | def _before_train(self): |
| 68 | # graph is finalized, OK to write it now. |
nothing calls this directly
no test coverage detected