Commit 454d8b38 authored by Karl Leswing's avatar Karl Leswing
Browse files

Lower learning rate for flaky test now that we have 1k epochs

parent 6ca79789

deepchem/utils/test/test_generator_evaluator.py

+1 −1
Original line number Diff line number Diff line
@@ -48,7 +48,7 @@ class TestGeneratorEvaluator(TestCase): 


    total_loss = ReduceMean(in_layers=entropies)



    tg = dc.models.TensorGraph(learning_rate=0.1)

    tg = dc.models.TensorGraph(learning_rate=0.01)

    for output in outputs:

      tg.add_output(output)

    tg.set_loss(total_loss)

CRA Git | Maintained and supported by SUSTech CRA and CCSE