Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "705487828ec5605721b93697c3d6b60d8a5bc9a0"
Commit 482784ab authored by Jing Li's avatar Jing Li Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 273233857
parent 7acce35d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/modeling/model_training_utils.py
View file @ 482784ab
...@@ -207,7 +207,7 @@ def run_customized_training_loop( ...@@ -207,7 +207,7 @@ def run_customized_training_loop(
'Checkpoint file %s found and restoring from ' 'Checkpoint file %s found and restoring from '
'initial checkpoint for core model.', init_checkpoint) 'initial checkpoint for core model.', init_checkpoint)
checkpoint = tf.train.Checkpoint(model=sub_model) checkpoint = tf.train.Checkpoint(model=sub_model)
checkpoint.restore(init_checkpoint).assert_consumed() checkpoint.restore(init_checkpoint).assert_existing_objects_matched()
logging.info('Loading from checkpoint file completed') logging.info('Loading from checkpoint file completed')
train_loss_metric = tf.keras.metrics.Mean( train_loss_metric = tf.keras.metrics.Mean(
......
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