Commit 0ec1a221 authored by Karl Leswing's avatar Karl Leswing
Browse files

test_overfit with higher learning rate

parent d69dfdc5

deepchem/models/tests/test_overfit.py

+1 −1
Original line number Diff line number Diff line
@@ -738,7 +738,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 
        n_tasks,

        n_feat,

        batch_size=batch_size,

        learning_rate=0.005,

        learning_rate=0.01,

        learning_rate_decay_time=1000,

        optimizer_type="adam",

        beta1=.9,

CRA Git | Maintained and supported by SUSTech CRA and CCSE