Skip to content

GitLab

Commit 9ac54873 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix regression 

PiperOrigin-RevId: 275603936
parent 49ba237d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/xlnet/training_utils.py
View file @ 9ac54873
...@@ -143,7 +143,7 @@ def train( ...@@ -143,7 +143,7 @@ def train(
transformer_xl=model.transformerxl_model) transformer_xl=model.transformerxl_model)
else: else:
checkpoint = tf.train.Checkpoint(model=model) checkpoint = tf.train.Checkpoint(model=model)
checkpoint.restore(init_checkpoint).assert_existing_objects_matched() checkpoint.restore(init_checkpoint)
model.optimizer = optimizer model.optimizer = optimizer
......
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