Commit 097727e4 authored by miaecle's avatar miaecle
Browse files

fix for failure

parent 16e8ae88
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -1005,7 +1005,7 @@ class TestOverfit(test_util.TensorFlowTestCase):
        mode="regression")

    # Fit trained model
    model.fit(dataset, nb_epoch=100)
    model.fit(dataset, nb_epoch=120)

    # Eval model on train
    scores = model.evaluate(dataset, [regression_metric])