From 630034b0dcced069ffd58670bb03e4276f93fb6a Mon Sep 17 00:00:00 2001 From: Marco Cognetta Date: Fri, 7 Jun 2024 22:30:26 +0900 Subject: [PATCH] enable dropout = 0.0 --- bindings/python/tests/bindings/test_models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/bindings/python/tests/bindings/test_models.py b/bindings/python/tests/bindings/test_models.py index 063698384..7b6746186 100644 --- a/bindings/python/tests/bindings/test_models.py +++ b/bindings/python/tests/bindings/test_models.py @@ -73,7 +73,6 @@ def test_dropout_zero(self): model = BPE(dropout=0.0) assert model.dropout == 0.0 - class TestWordPiece: def test_instantiate(self, bert_files): assert isinstance(WordPiece(), Model)