Skip to content

GitLab

Commit 92ea3959 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix unit test model config, we need to use kwargs to make dataclass correct. 

PiperOrigin-RevId: 327139660
parent 8da48573
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/tasks/masked_lm_test.py
View file @ 92ea3959
...@@ -29,7 +29,7 @@ class MLMTaskTest(tf.test.TestCase): ...@@ -29,7 +29,7 @@ class MLMTaskTest(tf.test.TestCase):
config = masked_lm.MaskedLMConfig( config = masked_lm.MaskedLMConfig(
init_checkpoint=self.get_temp_dir(), init_checkpoint=self.get_temp_dir(),
model=bert.PretrainerConfig( model=bert.PretrainerConfig(
encoders.EncoderConfig( encoder=encoders.EncoderConfig(
bert=encoders.BertEncoderConfig(vocab_size=30522, bert=encoders.BertEncoderConfig(vocab_size=30522,
num_layers=1)), num_layers=1)),
cls_heads=[ cls_heads=[
......
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