Skip to content

Commit

Permalink
Oops
Browse files Browse the repository at this point in the history
  • Loading branch information
AngledLuffa committed Dec 10, 2024
1 parent d4a81a7 commit f45919e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion stanza/tests/constituency/test_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,7 +274,7 @@ def test_early_dropout(self, wordvec_pretrain_file):
model = model.model
dropouts = [(name, module) for name, module in model.named_children() if isinstance(module, nn.Dropout)]
assert len(dropouts) > 0, "Didn't find any dropouts in the model!"
if all(module.p == 0.0 for module in dropouts):
if all(module.p == 0.0 for _, module in dropouts):
raise AssertionError("All dropouts were 0 after training even though early_dropout was set to -1")

def test_train_silver(self, wordvec_pretrain_file):
Expand Down

0 comments on commit f45919e

Please sign in to comment.