Update learning rate to make test more robust (snorkel-team#1715)

kili-technology · Oct 31, 2022 · a08509f · a08509f
1 parent d4509d0
commit a08509f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/classification/test_classifier_convergence.py b/test/classification/test_classifier_convergence.py
@@ -51,7 +51,7 @@ def test_convergence(self):
         model = MultitaskClassifier(tasks=[task1, task2])
 
         # Train
-        trainer = Trainer(lr=0.001, n_epochs=10, progress_bar=False)
+        trainer = Trainer(lr=0.0024, n_epochs=10, progress_bar=False)
         trainer.fit(model, dataloaders)
         scores = model.score(dataloaders)