Skip to content

Commit

Permalink
Increase the number of iterations for MatrixTest. (tensorflow#107)
Browse files Browse the repository at this point in the history
The number of iterations was reduced from 1000 to 300 during review, but that
turned out to be too low and the test now fails about 20% of the time.
After changing it back to 1000, the test succeeded at 50 out of 50 runs.
  • Loading branch information
judah authored May 9, 2017
1 parent a64af50 commit ff5f1cc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensorflow-ops/tests/MatrixTest.hs
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ fitMatrix = testCase "fitMatrix" $ TF.runSession $ do
diff = matx `TF.sub` (u `TF.matMul` v)
loss = reduceMean $ TF.square diff
trainStep <- gradientDescent 0.01 loss [u, v]
replicateM_ 300 (TF.run trainStep)
replicateM_ 1000 (TF.run trainStep)
(u',v') <- TF.run (u, v)
-- ones = u * v
liftIO $ assertAllClose (V.fromList ones) ((*) <$> u' <*> v')
Expand Down

0 comments on commit ff5f1cc

Please sign in to comment.