Home | History | Annotate | Download | only in estimator_batch

Lines Matching refs:model_dir

92     model_dir = tempfile.mkdtemp()
102 model_dir=model_dir,
115 model_dir = tempfile.mkdtemp()
125 model_dir=model_dir,
138 def _assert_checkpoint(self, model_dir, global_step):
139 reader = checkpoint_utils.load_checkpoint(model_dir)
149 model_dir = tempfile.mkdtemp()
159 model_dir=model_dir,
168 self._assert_checkpoint(est.model_dir, global_step=14)
180 model_dir = tempfile.mkdtemp()
190 model_dir=model_dir,
209 model_dir = tempfile.mkdtemp()
219 model_dir=model_dir,
240 export_path = os.path.join(model_dir, "export")