Commit 4457c1a8 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

remove use_remote_tpu as it is deprecated.

PiperOrigin-RevId: 277087491
parent 06f22a59
...@@ -96,7 +96,6 @@ def run_customized_training_loop( ...@@ -96,7 +96,6 @@ def run_customized_training_loop(
eval_steps=None, eval_steps=None,
metric_fn=None, metric_fn=None,
init_checkpoint=None, init_checkpoint=None,
use_remote_tpu=False,
custom_callbacks=None, custom_callbacks=None,
run_eagerly=False): run_eagerly=False):
"""Run BERT pretrain model training using low-level API. """Run BERT pretrain model training using low-level API.
...@@ -130,7 +129,6 @@ def run_customized_training_loop( ...@@ -130,7 +129,6 @@ def run_customized_training_loop(
after every epoch. after every epoch.
init_checkpoint: Optional checkpoint to load to `sub_model` returned by init_checkpoint: Optional checkpoint to load to `sub_model` returned by
`model_fn`. `model_fn`.
use_remote_tpu: Ignored, will be removed in the future.
custom_callbacks: A list of Keras Callbacks objects to run during custom_callbacks: A list of Keras Callbacks objects to run during
training. More specifically, `on_batch_begin()`, `on_batch_end()`, training. More specifically, `on_batch_begin()`, `on_batch_end()`,
methods are invoked during training. methods are invoked during training.
...@@ -145,8 +143,6 @@ def run_customized_training_loop( ...@@ -145,8 +143,6 @@ def run_customized_training_loop(
attribute or when required parameters are set to none. (2) eval args are attribute or when required parameters are set to none. (2) eval args are
not specified correctly. (3) metric_fn must be a callable if specified. not specified correctly. (3) metric_fn must be a callable if specified.
""" """
# TODO(bfontain): Remove use_remote_tpu once there are no models using it.
del use_remote_tpu
if _sentinel is not None: if _sentinel is not None:
raise ValueError('only call `run_customized_training_loop()` ' raise ValueError('only call `run_customized_training_loop()` '
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment