Commit 5544a2e8 authored by Zhenqin Wu's avatar Zhenqin Wu Committed by GitHub
Browse files

Update test_overfit.py

parent 5f1e81f6
Loading
Loading
Loading
Loading
+1 −1
Original line number Original line Diff line number Diff line
@@ -750,7 +750,7 @@ class TestOverfit(test_util.TensorFlowTestCase):
    # Eval model on train
    # Eval model on train
    scores = model.evaluate(dataset, [regression_metric])
    scores = model.evaluate(dataset, [regression_metric])


    assert scores[regression_metric.name] > .85
    assert scores[regression_metric.name] > .8


  def test_siamese_singletask_classification_overfit(self):
  def test_siamese_singletask_classification_overfit(self):
    """Test siamese singletask model overfits tiny data."""
    """Test siamese singletask model overfits tiny data."""