Commit 0baaeb85 authored by leswing's avatar leswing
Browse files

Lower learning rate

parent ff82a198

deepchem/models/tensorgraph/tests/test_tensor_graph.py

+1 −1
Original line number Diff line number Diff line
@@ -125,7 +125,7 @@ class TestTensorGraph(unittest.TestCase): 


    total_loss = ReduceMean(in_layers=losses)



    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