From d4ebd4e112034b4a429ab7f813d7e168e7bb63c3 Mon Sep 17 00:00:00 2001 From: Sijun He Date: Tue, 12 Jul 2022 16:28:28 +0800 Subject: [PATCH] speed up test (#18106) --- tests/models/nezha/test_modeling_nezha.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/nezha/test_modeling_nezha.py b/tests/models/nezha/test_modeling_nezha.py index 1d8017be95..1083ed0796 100644 --- a/tests/models/nezha/test_modeling_nezha.py +++ b/tests/models/nezha/test_modeling_nezha.py @@ -60,7 +60,8 @@ class NezhaModelTester: hidden_act="gelu", hidden_dropout_prob=0.1, attention_probs_dropout_prob=0.1, - max_position_embeddings=512, + max_position_embeddings=128, + max_relative_position=32, type_vocab_size=16, type_sequence_label_size=2, initializer_range=0.02,