Commit 5339fc1f authored by leswing's avatar leswing
Browse files

Use the new function

parent 66fa0ebb

deepchem/models/tensorgraph/tests/test_tensor_graph.py

+1 −1
Original line number Diff line number Diff line
@@ -178,7 +178,7 @@ class TestTensorGraph(unittest.TestCase): 
    tg = dc.models.TensorGraph(learning_rate=0.01, use_queue=False)

    tg.add_output(output)

    tg.set_loss(loss)

    global_step = tg._get_tf('GlobalStep')

    global_step = tg.get_global_step()



    def optimizer_function():

      starter_learning_rate = 0.1

CRA Git | Maintained and supported by SUSTech CRA and CCSE