Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
4457c1a8
Commit
4457c1a8
authored
Oct 28, 2019
by
Hongkun Yu
Committed by
A. Unique TensorFlower
Oct 28, 2019
Browse files
remove use_remote_tpu as it is deprecated.
PiperOrigin-RevId: 277087491
parent
06f22a59
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
4 deletions
+0
-4
official/modeling/model_training_utils.py
official/modeling/model_training_utils.py
+0
-4
No files found.
official/modeling/model_training_utils.py
View file @
4457c1a8
...
@@ -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()` '
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment