"torchvision/vscode:/vscode.git/clone" did not exist on "e79a74e1bfdf4e8e0275eeecea9063adba600ba0"
Commit 853d1a8b 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 093f3e1e
...@@ -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