(self)
| 516 | self.assertNotEmpty(tf.io.gfile.glob(done_file)) |
| 517 | |
| 518 | def test_no_eval_steps(self): |
| 519 | test_runner = TestRunner() |
| 520 | |
| 521 | checkpoint = tf.train.Checkpoint(model=test_runner.model) |
| 522 | checkpoint.save(os.path.join(self.model_dir, "ckpt")) |
| 523 | checkpoint_manager = tf.train.CheckpointManager( |
| 524 | checkpoint, |
| 525 | self.model_dir, |
| 526 | max_to_keep=None, |
| 527 | step_counter=test_runner.global_step) |
| 528 | test_controller = controller.Controller( |
| 529 | evaluator=test_runner, |
| 530 | global_step=test_runner.global_step, |
| 531 | checkpoint_manager=checkpoint_manager) |
| 532 | test_controller.evaluate() |
| 533 | |
| 534 | @parameterized.named_parameters( |
| 535 | ("_sync_checkpoint_saving", False), |
nothing calls this directly
no test coverage detected