Commit a3aa555a authored by miaecle's avatar miaecle
Browse files

little fix

parent 91777f4b

deepchem/models/tests/test_overfit.py

+1 −1
Original line number Diff line number Diff line
@@ -555,7 +555,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 
          optimizer_type="adam", beta1=.9, beta2=.999)



        # Fit trained model

        model.fit(dataset, nb_epoch=20)

        model.fit(dataset, nb_epoch=40)

        model.save()



        # Eval model on train

CRA Git | Maintained and supported by SUSTech CRA and CCSE