Commit b2f0404f authored by peastman's avatar peastman
Browse files

More dealing with flaky test case

parent abf9eeda

deepchem/metalearning/tests/test_maml.py

+1 −1
Original line number Diff line number Diff line
@@ -54,7 +54,7 @@ class TestMAML(unittest.TestCase): 
    learner = SineLearner()

    optimizer = dc.models.tensorgraph.optimizers.Adam(learning_rate=5e-3)

    maml = dc.metalearning.MAML(learner, meta_batch_size=4, optimizer=optimizer)

    maml.fit(7000)

    maml.fit(9000)



    # Test it out on some new tasks and see how it works.

CRA Git | Maintained and supported by SUSTech CRA and CCSE