Commit 18c60f83 authored by peastman's avatar peastman
Browse files

Fixed test failure

parent e0453f59

deepchem/metalearning/tests/test_maml.py

+1 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ class TestMAML(unittest.TestCase): 
    # Optimize it.



    learner = SineLearner()

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

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

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

    maml.fit(9000)

CRA Git | Maintained and supported by SUSTech CRA and CCSE