Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "d0a3058ba8049f7fd8214fc126e4bc0d8b5fa38b"
Commit dfce8c78 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 377088653
parent 8790769c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/tasks/sentence_prediction.py
View file @ dfce8c78
...@@ -206,10 +206,10 @@ class SentencePredictionTask(base_task.Task): ...@@ -206,10 +206,10 @@ class SentencePredictionTask(base_task.Task):
def initialize(self, model): def initialize(self, model):
"""Load a pretrained checkpoint (if exists) and then train from iter 0.""" """Load a pretrained checkpoint (if exists) and then train from iter 0."""
ckpt_dir_or_file = self.task_config.init_checkpoint ckpt_dir_or_file = self.task_config.init_checkpoint
if tf.io.gfile.isdir(ckpt_dir_or_file):
ckpt_dir_or_file = tf.train.latest_checkpoint(ckpt_dir_or_file)
if not ckpt_dir_or_file: if not ckpt_dir_or_file:
return return
if tf.io.gfile.isdir(ckpt_dir_or_file):
ckpt_dir_or_file = tf.train.latest_checkpoint(ckpt_dir_or_file)
pretrain2finetune_mapping = { pretrain2finetune_mapping = {
'encoder': model.checkpoint_items['encoder'], 'encoder': model.checkpoint_items['encoder'],
......
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