Wait till the global step has reached at least 'step'. Args: sess: A session. global_step: A Tensor. step: Int. The global step to reach.
(sess, global_step, step)
| 443 | |
| 444 | |
| 445 | def _wait_for_step(sess, global_step, step): |
| 446 | """Wait till the global step has reached at least 'step'. |
| 447 | |
| 448 | Args: |
| 449 | sess: A session. |
| 450 | global_step: A Tensor. |
| 451 | step: Int. The global step to reach. |
| 452 | """ |
| 453 | while True: |
| 454 | if training_util.global_step(sess, global_step) >= step: |
| 455 | break |
| 456 | time.sleep(1.0) |
| 457 | |
| 458 | |
| 459 | def train_step(sess, train_op, global_step, train_step_kwargs): |
no test coverage detected